Home
last modified time | relevance | path

Searched refs:bias_updates_gpu (Results 1 – 10 of 10) sorted by relevance

/dports/misc/darknet/darknet-darknet_yolo_v4_pre/src/
H A Ddeconvolutional_kernels.cu49 backward_bias(layer.bias_updates_gpu, layer.delta, layer.batch, layer.n, size); in backward_deconvolutional_layer_gpu()
85 cuda_pull_array(layer.bias_updates_gpu, layer.bias_updates, layer.n); in pull_deconvolutional_layer()
93 cuda_push_array(layer.bias_updates_gpu, layer.bias_updates, layer.n); in push_deconvolutional_layer()
100 axpy_ongpu(layer.n, learning_rate, layer.bias_updates_gpu, 1, layer.biases_gpu, 1); in update_deconvolutional_layer_gpu()
101 scal_ongpu(layer.n, momentum, layer.bias_updates_gpu, 1); in update_deconvolutional_layer_gpu()
H A Dconnected_layer.c130 l.bias_updates_gpu = cuda_make_array(l.bias_updates, outputs); in make_connected_layer()
288 cuda_pull_array(l.bias_updates_gpu, l.bias_updates, l.outputs); in pull_connected_layer()
302 cuda_push_array(l.bias_updates_gpu, l.bias_updates, l.outputs); in push_connected_layer()
318 scal_ongpu(l.outputs, 1.0 / loss_scale, l.bias_updates_gpu, 1); in update_connected_layer_gpu()
322 axpy_ongpu(l.outputs, learning_rate/batch, l.bias_updates_gpu, 1, l.biases_gpu, 1); in update_connected_layer_gpu()
323 scal_ongpu(l.outputs, momentum, l.bias_updates_gpu, 1); in update_connected_layer_gpu()
382 axpy_ongpu(l.outputs, 1, l.delta_gpu + i*l.outputs, 1, l.bias_updates_gpu, 1); in backward_connected_layer_gpu()
H A Dbatchnorm_layer.c55 layer.bias_updates_gpu = cuda_make_array(layer.bias_updates, c); in make_batchnorm_layer()
386 l.bias_updates_gpu, // output (should be FP32) in backward_batchnorm_layer_gpu()
393 backward_bias_gpu(l.bias_updates_gpu, l.delta_gpu, l.batch, l.out_c, l.out_w*l.out_h); in backward_batchnorm_layer_gpu()
407 fix_nan_and_inf(l.bias_updates_gpu, l.n); in backward_batchnorm_layer_gpu()
418 axpy_ongpu(l.c, learning_rate / batch, l.bias_updates_gpu, 1, l.biases_gpu, 1); in update_batchnorm_layer_gpu()
419 scal_ongpu(l.c, momentum, l.bias_updates_gpu, 1); in update_batchnorm_layer_gpu()
H A Dnetwork_kernels.cu417 cuda_pull_array(l.bias_updates_gpu, l.bias_updates, l.n); in pull_updates()
421 cuda_pull_array(l.bias_updates_gpu, l.bias_updates, l.outputs); in pull_updates()
429 cuda_push_array(l.bias_updates_gpu, l.bias_updates, l.n); in push_updates()
433 cuda_push_array(l.bias_updates_gpu, l.bias_updates, l.outputs); in push_updates()
531 cuda_push_array(l.bias_updates_gpu, base.bias_updates, l.n); in distribute_updates()
535 cuda_push_array(l.bias_updates_gpu, base.bias_updates, l.outputs); in distribute_updates()
H A Dlocal_layer.c77 l.bias_updates_gpu = cuda_make_array(l.bias_updates, l.outputs); in make_local_layer()
219 axpy_ongpu(l.outputs, 1, l.delta_gpu + i*l.outputs, 1, l.bias_updates_gpu, 1); in backward_local_layer_gpu()
260 axpy_ongpu(l.outputs, learning_rate/batch, l.bias_updates_gpu, 1, l.biases_gpu, 1); in update_local_layer_gpu()
261 scal_ongpu(l.outputs, momentum, l.bias_updates_gpu, 1); in update_local_layer_gpu()
H A Dconvolutional_kernels.cu657 backward_bias_gpu(l.bias_updates_gpu, l.delta_gpu, l.batch, l.n, l.out_w*l.out_h); in backward_convolutional_layer_gpu()
724 l.bias_updates_gpu, // output (should be FP32) in backward_convolutional_layer_gpu()
1159 cuda_pull_array_async(l.bias_updates_gpu, l.bias_updates, l.n); in pull_convolutional_layer()
1183 cuda_push_array(l.bias_updates_gpu, l.bias_updates, l.n); in push_convolutional_layer()
1244 if (l.bias_updates_gpu && l.n > 0) scal_ongpu(l.n, 1.0 / loss_scale, l.bias_updates_gpu, 1); in update_convolutional_layer_gpu()
1263 …adam_update_gpu(l.biases_gpu, l.bias_updates_gpu, l.bias_m_gpu, l.bias_v_gpu, l.B1, l.B2, l.eps, d… in update_convolutional_layer_gpu()
1276 axpy_ongpu(l.n, learning_rate / batch, l.bias_updates_gpu, 1, l.biases_gpu, 1); in update_convolutional_layer_gpu()
1277 scal_ongpu(l.n, momentum, l.bias_updates_gpu, 1); in update_convolutional_layer_gpu()
H A Dlayer.c186 if (l.bias_updates_gpu) cuda_free(l.bias_updates_gpu), l.bias_updates_gpu = NULL; in free_layer_custom()
H A Ddeconvolutional_layer.c92 l.bias_updates_gpu = cuda_make_array(l.bias_updates, n); in make_deconvolutional_layer()
H A Dconvolutional_layer.c570 l.bias_updates_gpu = l.share_layer->bias_updates_gpu; in make_convolutional_layer()
580 if (train) l.bias_updates_gpu = cuda_make_array(l.bias_updates, n); in make_convolutional_layer()
/dports/misc/darknet/darknet-darknet_yolo_v4_pre/include/
H A Ddarknet.h592 float * bias_updates_gpu; member