Home
last modified time | relevance | path

Searched refs:a_avg_gpu (Results 1 – 4 of 4) sorted by relevance

/dports/misc/darknet/darknet-darknet_yolo_v4_pre/src/
H A Dconvolutional_kernels.cu938 …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 Dlayer.c173 if (l.a_avg_gpu) cuda_free(l.a_avg_gpu); in free_layer_custom()
H A Dconvolutional_layer.c638 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 Ddarknet.h612 float *a_avg_gpu; member