16 lines
412 B
C
16 lines
412 B
C
|
#pragma once
|
||
|
|
|
||
|
|
void timer_start(int i);
|
||
|
|
|
||
|
|
double timer_stop(int i);
|
||
|
|
|
||
|
|
void check_convolution(float *input, float *output, float *filter,
|
||
|
|
int N, int C, int H, int W, int K, int R, int S,
|
||
|
|
int pad, int dilation, int stride);
|
||
|
|
|
||
|
|
void alloc_tensor(float **t, int D0, int D1, int D2, int D3);
|
||
|
|
|
||
|
|
void rand_tensor(float *t, int D0, int D1, int D2, int D3);
|
||
|
|
|
||
|
|
void zero_tensor(float *t, int D0, int D1, int D2, int D3);
|