/dports/misc/darknet/darknet-darknet_yolo_v4_pre/src/ |
H A D | deconvolutional_kernels.cu | 49 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 D | connected_layer.c | 130 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 D | batchnorm_layer.c | 55 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 D | network_kernels.cu | 417 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 D | local_layer.c | 77 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 D | convolutional_kernels.cu | 657 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 D | layer.c | 186 if (l.bias_updates_gpu) cuda_free(l.bias_updates_gpu), l.bias_updates_gpu = NULL; in free_layer_custom()
|
H A D | deconvolutional_layer.c | 92 l.bias_updates_gpu = cuda_make_array(l.bias_updates, n); in make_deconvolutional_layer()
|
H A D | convolutional_layer.c | 570 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 D | darknet.h | 592 float * bias_updates_gpu; member
|