Home
last modified time | relevance | path

Searched refs:weights_gpu16 (Results 1 – 5 of 5) sorted by relevance

/dports/misc/darknet/darknet-darknet_yolo_v4_pre/src/
H A Dnetwork_kernels.cu312 if (l.type == CONVOLUTIONAL && l.weights_gpu && l.weights_gpu16) { in forward_backward_network_gpu()
314 cuda_convert_f32_to_f16(l.weights_gpu, l.nweights, l.weights_gpu16); in forward_backward_network_gpu()
322 else if (l.type == CONV_LSTM && l.wf->weights_gpu && l.wf->weights_gpu16) { in forward_backward_network_gpu()
329 cuda_convert_f32_to_f16(l.wf->weights_gpu, l.wf->nweights, l.wf->weights_gpu16); in forward_backward_network_gpu()
330 cuda_convert_f32_to_f16(l.wi->weights_gpu, l.wi->nweights, l.wi->weights_gpu16); in forward_backward_network_gpu()
331 cuda_convert_f32_to_f16(l.wg->weights_gpu, l.wg->nweights, l.wg->weights_gpu16); in forward_backward_network_gpu()
332 cuda_convert_f32_to_f16(l.wo->weights_gpu, l.wo->nweights, l.wo->weights_gpu16); in forward_backward_network_gpu()
333 cuda_convert_f32_to_f16(l.uf->weights_gpu, l.uf->nweights, l.uf->weights_gpu16); in forward_backward_network_gpu()
334 cuda_convert_f32_to_f16(l.ui->weights_gpu, l.ui->nweights, l.ui->weights_gpu16); in forward_backward_network_gpu()
335 cuda_convert_f32_to_f16(l.ug->weights_gpu, l.ug->nweights, l.ug->weights_gpu16); in forward_backward_network_gpu()
[all …]
H A Dlayer.c183 if (l.weights_gpu16) cuda_free(l.weights_gpu16), l.weights_gpu16 = NULL; in free_layer_custom()
H A Dconvolutional_kernels.cu463 l.weights_gpu16, in forward_convolutional_layer_gpu()
775 l.weights_gpu16, //l.weights_gpu, in backward_convolutional_layer_gpu()
1178 cuda_convert_f32_to_f16(l.weights_gpu, l.nweights, l.weights_gpu16); in push_convolutional_layer()
H A Dconvolutional_layer.c567 l.weights_gpu16 = l.share_layer->weights_gpu16; in make_convolutional_layer()
576 l.weights_gpu16 = cuda_make_array(NULL, l.nweights / 2 + 1); in make_convolutional_layer()
/dports/misc/darknet/darknet-darknet_yolo_v4_pre/include/
H A Ddarknet.h588 float * weights_gpu16; member