Searched refs:weights_gpu16 (Results 1 – 5 of 5) sorted by relevance
/dports/misc/darknet/darknet-darknet_yolo_v4_pre/src/ |
H A D | network_kernels.cu | 312 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 D | layer.c | 183 if (l.weights_gpu16) cuda_free(l.weights_gpu16), l.weights_gpu16 = NULL; in free_layer_custom()
|
H A D | convolutional_kernels.cu | 463 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 D | convolutional_layer.c | 567 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 D | darknet.h | 588 float * weights_gpu16; member
|