| D | eigen_benchmark_cpu_test.cc | 124 #define BM_SPATIAL_NAME(prefix, NT, N, H, W, C, FC, FH, FW) \ argument
125 BM_##prefix##_CPU_##NT##T_in_##N##_##H##_##W##_##C##_f_##FC##_##FH##_##FW
127 #define BM_SpatialConvolution(NT, N, H, W, C, FC, FH, FW, LABEL) \ argument
128 static void BM_SPATIAL_NAME(SpatialConvolution, NT, N, H, W, C, FC, FH, \
131 SpatialConvolution(iters, NT, N, H, W, C, FC, FH, FW); \
133 BENCHMARK(BM_SPATIAL_NAME(SpatialConvolution, NT, N, H, W, C, FC, FH, FW))
135 #define BM_SpatialConvolutionBwdInput(NT, N, H, W, C, FC, FH, FW, LABEL) \ argument
136 static void BM_SPATIAL_NAME(SpatialConvolutionBwdInput, NT, N, H, W, C, FC, \
139 SpatialConvolutionBackwardInput(iters, NT, N, H, W, C, FC, FH, FW); \
142 BM_SPATIAL_NAME(SpatialConvolutionBwdInput, NT, N, H, W, C, FC, FH, FW))
[all …]
|