maxpool_layer_kernels.cu File Reference

#include "cuda_runtime.h"
#include "curand.h"
#include "cublas_v2.h"
#include "maxpool_layer.h"
#include "cuda.h"

Include dependency graph for maxpool_layer_kernels.cu:

Go to the source code of this file.

Functions
__global__ void	forward_maxpool_layer_kernel (int n, int in_h, int in_w, int in_c, int stride, int size, int pad, float *input, float *output, int *indexes)
__global__ void	backward_maxpool_layer_kernel (int n, int in_h, int in_w, int in_c, int stride, int size, int pad, float *delta, float *prev_delta, int *indexes)
void	forward_maxpool_layer_gpu (maxpool_layer layer, network net)
void	backward_maxpool_layer_gpu (maxpool_layer layer, network net)

Function Documentation

backward_maxpool_layer_gpu()

void backward_maxpool_layer_gpu	(	maxpool_layer	layer,
		network	net
	)

Definition at line 99 of file maxpool_layer_kernels.cu.

backward_maxpool_layer_kernel()

__global__ void backward_maxpool_layer_kernel	(	int	n,
		int	in_h,
		int	in_w,
		int	in_c,
		int	stride,
		int	size,
		int	pad,
		float *	delta,
		float *	prev_delta,
		int *	indexes
	)

Definition at line 50 of file maxpool_layer_kernels.cu.

forward_maxpool_layer_gpu()

void forward_maxpool_layer_gpu	(	maxpool_layer	layer,
		network	net
	)

Definition at line 87 of file maxpool_layer_kernels.cu.

forward_maxpool_layer_kernel()

__global__ void forward_maxpool_layer_kernel	(	int	n,
		int	in_h,
		int	in_w,
		int	in_c,
		int	stride,
		int	size,
		int	pad,
		float *	input,
		float *	output,
		int *	indexes
	)

Definition at line 10 of file maxpool_layer_kernels.cu.