![]() |
darknet
v3
|
#include "cuda_runtime.h"
#include "curand.h"
#include "cublas_v2.h"
#include "col2im.h"
#include "cuda.h"
Go to the source code of this file.
Functions | |
__global__ void | col2im_gpu_kernel (const int n, const float *data_col, 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_im) |
void | col2im_gpu (float *data_col, int channels, int height, int width, int ksize, int stride, int pad, float *data_im) |
void col2im_gpu | ( | float * | data_col, |
int | channels, | ||
int | height, | ||
int | width, | ||
int | ksize, | ||
int | stride, | ||
int | pad, | ||
float * | data_im | ||
) |
Definition at line 44 of file col2im_kernels.cu.
__global__ void col2im_gpu_kernel | ( | const int | n, |
const float * | data_col, | ||
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_im | ||
) |
Definition at line 13 of file col2im_kernels.cu.