dropout_layer_kernels.cu File Reference

#include "cuda_runtime.h"
#include "curand.h"
#include "cublas_v2.h"
#include "dropout_layer.h"
#include "cuda.h"
#include "utils.h"

Include dependency graph for dropout_layer_kernels.cu:

Go to the source code of this file.

Functions
__global__ void	yoloswag420blazeit360noscope (float *input, int size, float *rand, float prob, float scale)
void	forward_dropout_layer_gpu (dropout_layer layer, network net)
void	backward_dropout_layer_gpu (dropout_layer layer, network net)

Function Documentation

backward_dropout_layer_gpu()

void backward_dropout_layer_gpu	(	dropout_layer	layer,
		network	net
	)

Definition at line 34 of file dropout_layer_kernels.cu.

forward_dropout_layer_gpu()

void forward_dropout_layer_gpu	(	dropout_layer	layer,
		network	net
	)

Definition at line 17 of file dropout_layer_kernels.cu.

yoloswag420blazeit360noscope()

__global__ void yoloswag420blazeit360noscope	(	float *	input,
		int	size,
		float *	rand,
		float	prob,
		float	scale
	)

Definition at line 11 of file dropout_layer_kernels.cu.