2023-02-01 20:04:35 +09:00
|
|
|
#pragma once
|
|
|
|
|
|
|
|
double get_current_time();
|
|
|
|
|
|
|
|
void alloc_tensor(float **m, int N, int C, int H, int W);
|
|
|
|
|
|
|
|
void rand_tensor(float *m, int N, int C, int H, int W);
|
|
|
|
|
|
|
|
void zero_tensor(float *m, int N, int C, int H, int W);
|
|
|
|
|
|
|
|
void print_tensor(float *m, int N, int C, int H, int W);
|
|
|
|
|
2023-02-14 01:23:28 +09:00
|
|
|
void check_convolution(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);
|