Searched refs:a_avg_gpu (Results 1 – 4 of 4) sorted by relevance
/dports/misc/darknet/darknet-darknet_yolo_v4_pre/src/ |
H A D | convolutional_kernels.cu | 938 …ted_activation_kernel(float alpha, float *output, float *gt_gpu, float *a_avg_gpu, int size, int c… in assisted_activation_kernel() argument 946 output[xy + size*(c + channels*b)] += alpha * gt_gpu[i] * a_avg_gpu[i]; in assisted_activation_kernel() 954 void assisted_activation_gpu(float alpha, float *output, float *gt_gpu, float *a_avg_gpu, int size,… in assisted_activation_gpu() argument 958 …< <num_blocks, BLOCK, 0, get_cuda_stream() >> > (alpha, output, gt_gpu, a_avg_gpu, size, channels,… in assisted_activation_gpu() 962 …ed_activation2_kernel(float alpha, float *output, float *gt_gpu, float *a_avg_gpu, int size, int c… in assisted_activation2_kernel() argument 978 void assisted_activation2_gpu(float alpha, float *output, float *gt_gpu, float *a_avg_gpu, int size… in assisted_activation2_gpu() argument 982 …< <num_blocks, BLOCK, 0, get_cuda_stream() >> > (alpha, output, gt_gpu, a_avg_gpu, size, channels,… in assisted_activation2_gpu() 1064 calc_avg_activation_gpu(l.output_gpu, l.a_avg_gpu, l.out_w * l.out_h, l.out_c, l.batch); in assisted_excitation_forward_gpu() 1071 …assisted_activation_gpu(alpha, l.output_gpu, l.gt_gpu, l.a_avg_gpu, l.out_w * l.out_h, l.out_c, l.… in assisted_excitation_forward_gpu()
|
H A D | layer.c | 173 if (l.a_avg_gpu) cuda_free(l.a_avg_gpu); in free_layer_custom()
|
H A D | convolutional_layer.c | 638 l.a_avg_gpu = cuda_make_array(NULL, size); in make_convolutional_layer() 821 cuda_free(l->a_avg_gpu); in resize_convolutional_layer() 825 l->a_avg_gpu = cuda_make_array(NULL, size); in resize_convolutional_layer()
|
/dports/misc/darknet/darknet-darknet_yolo_v4_pre/include/ |
H A D | darknet.h | 612 float *a_avg_gpu; member
|