Darknet/YOLO v3.0-149-gb11c9d5
Object Detection Framework
 
Loading...
Searching...
No Matches
dropout_layer_kernels.cu File Reference

Functions

__global__ void backward_dropblock_kernel (float *pass, float *delta, int size)
 
void backward_dropout_layer_gpu (Darknet::Layer &l, Darknet::NetworkState state)
 
__global__ void dropblock_fast_kernel (float *rand, float prob, int w, int h, int spatial, int filters, int batch, int block_size, float *drop_blocks_scale, float *output)
 
void forward_dropout_layer_gpu (Darknet::Layer &l, Darknet::NetworkState state)
 
__global__ void scale_dropblock_kernel (float *output, int size, int outputs, float *drop_blocks_scale)
 
__global__ void set_scales_dropblock_kernel (float *drop_blocks_scale, int block_size_w, int block_size_h, int outputs, int batch)
 
__global__ void yoloswag420blazeit360noscope (float *input, int size, float *rand, float prob, float scale)
 

Function Documentation

◆ backward_dropblock_kernel()

__global__ void backward_dropblock_kernel ( float *  pass,
float *  delta,
int  size 
)
Here is the caller graph for this function:

◆ backward_dropout_layer_gpu()

void backward_dropout_layer_gpu ( Darknet::Layer l,
Darknet::NetworkState  state 
)
Here is the call graph for this function:
Here is the caller graph for this function:

◆ dropblock_fast_kernel()

__global__ void dropblock_fast_kernel ( float *  rand,
float  prob,
int  w,
int  h,
int  spatial,
int  filters,
int  batch,
int  block_size,
float *  drop_blocks_scale,
float *  output 
)
Here is the caller graph for this function:

◆ forward_dropout_layer_gpu()

void forward_dropout_layer_gpu ( Darknet::Layer l,
Darknet::NetworkState  state 
)
Here is the call graph for this function:
Here is the caller graph for this function:

◆ scale_dropblock_kernel()

__global__ void scale_dropblock_kernel ( float *  output,
int  size,
int  outputs,
float *  drop_blocks_scale 
)
Here is the caller graph for this function:

◆ set_scales_dropblock_kernel()

__global__ void set_scales_dropblock_kernel ( float *  drop_blocks_scale,
int  block_size_w,
int  block_size_h,
int  outputs,
int  batch 
)
Here is the caller graph for this function:

◆ yoloswag420blazeit360noscope()

__global__ void yoloswag420blazeit360noscope ( float *  input,
int  size,
float *  rand,
float  prob,
float  scale 
)
Here is the caller graph for this function: