Searched refs:drop_blocks_scale (Results 1 – 4 of 4) sorted by relevance
/dports/misc/darknet/darknet-darknet_yolo_v4_pre/src/ |
H A D | dropout_layer_kernels.cu | 15 … int h, int spatial, int filters, int batch, int block_size, float *drop_blocks_scale, float *outp… in dropblock_fast_kernel() argument 80 if (id == 0 && drop_blocks_scale) { in dropblock_fast_kernel() 81 atomicAdd(&drop_blocks_scale[b], block_square_size); in dropblock_fast_kernel() 87 __global__ void set_scales_dropblock_kernel(float *drop_blocks_scale, int block_size_w, int block_s… in set_scales_dropblock_kernel() argument 93 const float prob = drop_blocks_scale[index] / (float)outputs; in set_scales_dropblock_kernel() 95 drop_blocks_scale[index] = scale; in set_scales_dropblock_kernel() 98 …lobal__ void scale_dropblock_kernel(float *output, int size, int outputs, float *drop_blocks_scale) in scale_dropblock_kernel() argument 104 output[index] *= drop_blocks_scale[b]; in scale_dropblock_kernel()
|
H A D | dropout_layer.c | 37 l.drop_blocks_scale = cuda_make_array_pinned(l.rand, l.batch); in make_dropout_layer() 58 cudaFreeHost(l->drop_blocks_scale); in resize_dropout_layer() 59 l->drop_blocks_scale = cuda_make_array_pinned(l->rand, l->batch); in resize_dropout_layer()
|
H A D | layer.c | 59 if (l.drop_blocks_scale) cuda_free_host(l.drop_blocks_scale); in free_layer_custom()
|
/dports/misc/darknet/darknet-darknet_yolo_v4_pre/include/ |
H A D | darknet.h | 606 float * drop_blocks_scale; member
|