avgpool_layer_kernels.cu File Reference

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

Include dependency graph for avgpool_layer_kernels.cu:

Go to the source code of this file.

Functions
__global__ void	forward_avgpool_layer_kernel (int n, int w, int h, int c, float *input, float *output)
__global__ void	backward_avgpool_layer_kernel (int n, int w, int h, int c, float *in_delta, float *out_delta)
void	forward_avgpool_layer_gpu (avgpool_layer layer, network net)
void	backward_avgpool_layer_gpu (avgpool_layer layer, network net)

Function Documentation

backward_avgpool_layer_gpu()

void backward_avgpool_layer_gpu	(	avgpool_layer	layer,
		network	net
	)

Definition at line 54 of file avgpool_layer_kernels.cu.

backward_avgpool_layer_kernel()

__global__ void backward_avgpool_layer_kernel	(	int	n,
		int	w,
		int	h,
		int	c,
		float *	in_delta,
		float *	out_delta
	)

Definition at line 29 of file avgpool_layer_kernels.cu.

forward_avgpool_layer_gpu()

void forward_avgpool_layer_gpu	(	avgpool_layer	layer,
		network	net
	)

Definition at line 46 of file avgpool_layer_kernels.cu.

forward_avgpool_layer_kernel()

__global__ void forward_avgpool_layer_kernel	(	int	n,
		int	w,
		int	h,
		int	c,
		float *	input,
		float *	output
	)

Definition at line 10 of file avgpool_layer_kernels.cu.