23 lines
781 B
Plaintext
23 lines
781 B
Plaintext
#include <cstdlib>
|
|
|
|
#include "convolution.h"
|
|
|
|
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) {
|
|
// TODO
|
|
return;
|
|
}
|
|
|
|
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) {
|
|
// TODO
|
|
}
|
|
|
|
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) {
|
|
// TODO
|
|
return;
|
|
} |