#include "cuda_runtime.h"
#include "curand.h"
#include "cublas_v2.h"
#include "convolutional_layer.h"
#include "batchnorm_layer.h"
#include "gemm.h"
#include "blas.h"
#include "im2col.h"
#include "col2im.h"
#include "utils.h"
#include "cuda.h"

Include dependency graph for convolutional_kernels.cu:

Functions
__global__ void	binarize_kernel (float x, int n, float binary)

void	binarize_gpu (float x, int n, float binary)

__global__ void	binarize_input_kernel (float input, int n, int size, float binary)

void	binarize_input_gpu (float input, int n, int size, float binary)

__global__ void	binarize_weights_kernel (float weights, int n, int size, float binary)

void	binarize_weights_gpu (float weights, int n, int size, float binary)

void	forward_convolutional_layer_gpu (convolutional_layer l, network net)

__global__ void	smooth_kernel (float x, int n, int w, int h, int c, int size, float rate, float delta)

void	smooth_layer (layer l, int size, float rate)

void	backward_convolutional_layer_gpu (convolutional_layer l, network net)

void	pull_convolutional_layer (layer l)

void	push_convolutional_layer (layer l)

void	update_convolutional_layer_gpu (layer l, update_args a)

Function Documentation

◆ backward_convolutional_layer_gpu()

void backward_convolutional_layer_gpu	(	convolutional_layer	l,
		network	net
	)

Definition at line 179 of file convolutional_kernels.cu.

◆ binarize_gpu()

void binarize_gpu	(	float *	x,
		int	n,
		float *	binary
	)

Definition at line 23 of file convolutional_kernels.cu.

◆ binarize_input_gpu()

void binarize_input_gpu	(	float *	input,
		int	n,
		int	size,
		float *	binary
	)

Definition at line 44 of file convolutional_kernels.cu.

◆ binarize_input_kernel()

__global__ void binarize_input_kernel	(	float *	input,
		int	n,
		int	size,
		float *	binary
	)

Definition at line 29 of file convolutional_kernels.cu.

◆ binarize_kernel()

__global__ void binarize_kernel	(	float *	x,
		int	n,
		float *	binary
	)

Definition at line 16 of file convolutional_kernels.cu.

◆ binarize_weights_gpu()

void binarize_weights_gpu	(	float *	weights,
		int	n,
		int	size,
		float *	binary
	)

Definition at line 67 of file convolutional_kernels.cu.

◆ binarize_weights_kernel()

__global__ void binarize_weights_kernel	(	float *	weights,
		int	n,
		int	size,
		float *	binary
	)

Definition at line 51 of file convolutional_kernels.cu.

◆ forward_convolutional_layer_gpu()

void forward_convolutional_layer_gpu	(	convolutional_layer	l,
		network	net
	)

Definition at line 73 of file convolutional_kernels.cu.

◆ pull_convolutional_layer()

void pull_convolutional_layer ( layer l )

Definition at line 273 of file convolutional_kernels.cu.

◆ push_convolutional_layer()

void push_convolutional_layer ( layer l )

Definition at line 286 of file convolutional_kernels.cu.

◆ smooth_kernel()

__global__ void smooth_kernel	(	float *	x,
		int	n,
		int	w,
		int	h,
		int	c,
		int	size,
		float	rate,
		float *	delta
	)

Definition at line 137 of file convolutional_kernels.cu.

◆ smooth_layer()

void smooth_layer	(	layer	l,
		int	size,
		float	rate
	)

Definition at line 167 of file convolutional_kernels.cu.

◆ update_convolutional_layer_gpu()

void update_convolutional_layer_gpu	(	layer	l,
		update_args	a
	)

Definition at line 299 of file convolutional_kernels.cu.

Functions