2023-02-01 20:04:35 +09:00
|
|
|
#pragma once
|
|
|
|
|
|
|
|
|
|
#include <cstddef>
|
|
|
|
|
|
2023-02-14 01:23:28 +09:00
|
|
|
void convolution_cpu(float *I, float *F, float *O, int N, int C, int H, int W,
|
|
|
|
|
int K, int R, int S, int pad_h, int pad_w, int stride_h,
|
|
|
|
|
int stride_w, int dilation_h, int dilation_w);
|
|
|
|
|
void convolution_gpu(float *I, float *F, float *O, int N, int C, int H, int W,
|
|
|
|
|
int K, int R, int S, int pad_h, int pad_w, int stride_h,
|
|
|
|
|
int stride_w, int dilation_h, int dilation_w);
|
|
|
|
|
void convolution_cudnn(float *I, float *F, float *O, int N, int C, int H, int W,
|
|
|
|
|
int K, int R, int S, int pad_h, int pad_w, int stride_h,
|
|
|
|
|
int stride_w, int dilation_h, int dilation_w);
|
2023-02-01 20:04:35 +09:00
|
|
|
|
2023-02-14 01:23:28 +09:00
|
|
|
void convolution_cpu_initialize(int N, int C, int H, int W, int K, int R, int S,
|
|
|
|
|
int pad_h, int pad_w, int stride_h,
|
|
|
|
|
int stride_w, int dilation_h, int dilation_w);
|
|
|
|
|
void convolution_gpu_initialize(int N, int C, int H, int W, int K, int R, int S,
|
|
|
|
|
int pad_h, int pad_w, int stride_h,
|
|
|
|
|
int stride_w, int dilation_h, int dilation_w);
|
|
|
|
|
void convolution_cudnn_initialize(int N, int C, int H, int W, int K, int R,
|
|
|
|
|
int S, int pad_h, int pad_w, int stride_h,
|
|
|
|
|
int stride_w, int dilation_h, int dilation_w);
|
2023-02-01 20:04:35 +09:00
|
|
|
|
2023-02-14 01:23:28 +09:00
|
|
|
void convolution_cpu_finalize(int N, int C, int H, int W, int K, int R, int S,
|
|
|
|
|
int pad_h, int pad_w, int stride_h, int stride_w,
|
|
|
|
|
int dilation_h, int dilation_w);
|
|
|
|
|
void convolution_gpu_finalize(int N, int C, int H, int W, int K, int R, int S,
|
|
|
|
|
int pad_h, int pad_w, int stride_h, int stride_w,
|
|
|
|
|
int dilation_h, int dilation_w);
|
|
|
|
|
void convolution_cudnn_finalize(int N, int C, int H, int W, int K, int R, int S,
|
|
|
|
|
int pad_h, int pad_w, int stride_h,
|
|
|
|
|
int stride_w, int dilation_h, int dilation_w);
|
