forked from jiuyuan/InfiniTensor
16 lines
646 B
C
16 lines
646 B
C
|
#pragma once
|
||
|
|
namespace infini {
|
||
|
|
namespace opTimer {
|
||
|
|
double getPerfConvMkl(int n, int c, int h, int w, int f, int r, int s, int padh,
|
||
|
|
int padw, int strideh, int stridew, int dilationh,
|
||
|
|
int dilationw, int group);
|
||
|
|
|
||
|
|
double getPerfConvTransposed2dMkl(int n, int c, int h, int w, int f, int r,
|
||
|
|
int s, int padh, int padw, int strideh,
|
||
|
|
int stridew, int dilationh, int dilationw,
|
||
|
|
int oph, int opw, int group);
|
||
|
|
|
||
|
|
double getPerfMatmulMkl(int b, int m, int n, int k);
|
||
|
|
} // namespace opTimer
|
||
|
|
} // namespace infini
|