forked from jiuyuan/InfiniTensor
11 lines
416 B
C++
11 lines
416 B
C++
#pragma once
|
|
namespace infini {
|
|
namespace opTimer {
|
|
double getPerfConvCnnl(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,
|
|
const char *name);
|
|
double getPerfMatmulCnnl(int b, int m, int n, int k, const char *name);
|
|
} // namespace opTimer
|
|
} // namespace infini
|