forked from jiuyuan/InfiniTensor
14 lines
611 B
C
14 lines
611 B
C
|
#pragma once
|
||
|
|
|
||
|
|
namespace infini {
|
||
|
|
|
||
|
|
void conv2dreduce_kernel(float *input, float *bias, float *output, bool PReLU,
|
||
|
|
int n, int h, int w, int f, int r, int s, int oh,
|
||
|
|
int ow, int ph, int pw, int sh, int sw, int dh,
|
||
|
|
int dw);
|
||
|
|
|
||
|
|
void convTranspose2dreduce_kernel(float *input, float *bias, float *output,
|
||
|
|
int act, int n, int h, int w, int f, int r,
|
||
|
|
int s, int oh, int ow, int ph, int pw, int sh,
|
||
|
|
int sw, int dh, int dw);
|
||
|
|
} // namespace infini
|