| D | eigen_benchmark_cpu_test.cc | 120 #define BM_SPATIAL_NAME(prefix, NT, N, H, W, C, FC, FH, FW) \ argument
121 BM_##prefix##_CPU_##NT##T_in_##N##_##H##_##W##_##C##_f_##FC##_##FH##_##FW
123 #define BM_SpatialConvolution(NT, N, H, W, C, FC, FH, FW, LABEL) \ argument
124 static void BM_SPATIAL_NAME(SpatialConvolution, NT, N, H, W, C, FC, FH, \
127 SpatialConvolution(state, NT, N, H, W, C, FC, FH, FW); \
129 BENCHMARK(BM_SPATIAL_NAME(SpatialConvolution, NT, N, H, W, C, FC, FH, FW))
131 #define BM_SpatialConvolutionBwdInput(NT, N, H, W, C, FC, FH, FW, LABEL) \ argument
132 static void BM_SPATIAL_NAME(SpatialConvolutionBwdInput, NT, N, H, W, C, FC, \
135 SpatialConvolutionBackwardInput(state, NT, N, H, W, C, FC, FH, FW); \
138 BM_SPATIAL_NAME(SpatialConvolutionBwdInput, NT, N, H, W, C, FC, FH, FW))
[all …]
|