im2col_kernels.cu File Reference

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

Include dependency graph for im2col_kernels.cu:

Go to the source code of this file.

Functions
__global__ void	im2col_gpu_kernel (const int n, const float *data_im, const int height, const int width, const int ksize, const int pad, const int stride, const int height_col, const int width_col, float *data_col)
void	im2col_gpu (float *im, int channels, int height, int width, int ksize, int stride, int pad, float *data_col)

Function Documentation

im2col_gpu()

void im2col_gpu	(	float *	im,
		int	channels,
		int	height,
		int	width,
		int	ksize,
		int	stride,
		int	pad,
		float *	data_col
	)

Definition at line 48 of file im2col_kernels.cu.

im2col_gpu_kernel()

__global__ void im2col_gpu_kernel	(	const int	n,
		const float *	data_im,
		const int	height,
		const int	width,
		const int	ksize,
		const int	pad,
		const int	stride,
		const int	height_col,
		const int	width_col,
		float *	data_col
	)

Definition at line 13 of file im2col_kernels.cu.