| /external/tensorflow/tensorflow/core/kernels/ |
| D | eigen_benchmark_cpu_test.cc | 124 #define BM_SPATIAL_NAME(prefix, NT, N, H, W, C, FC, FH, FW) \ argument
127 #define BM_SpatialConvolution(NT, N, H, W, C, FC, FH, FW, LABEL) \ argument
135 #define BM_SpatialConvolutionBwdInput(NT, N, H, W, C, FC, FH, FW, LABEL) \ argument
144 #define BM_SpatialConvolutionBwdKernel(NT, N, H, W, C, FC, FH, FW, LABEL) \ argument
153 #define BM_SpatialConvolutions(N, H, W, C, FC, FH, FW, LABEL) \ argument
159 #define BM_SpatialConvolutionsBwdInput(N, H, W, C, FC, FH, FW, LABEL) \ argument
165 #define BM_SpatialConvolutionsBwdKernel(N, H, W, C, FC, FH, FW, LABEL) \ argument
352 #define BM_CUBOID_NAME(p, NT, N, H, W, P, C, FC, FH, FW, FP) \ argument
356 #define BM_CuboidConvolution(NT, N, H, W, P, C, FC, FH, FW, FP, LABEL) \ argument
365 #define BM_CuboidConvolutionBwdInput(NT, N, H, W, P, C, FC, FH, FW, FP, LABEL) \ argument
[all …]
|
| D | mkl_conv_ops_test.cc | 306 #define BM_NAME(p, type, N, H, W, C, FC, FH, FW) \ argument
312 #define BM_Conv2DT(kind, N, H, W, C, FC, FH, FW, type, LABEL) \ argument
327 #define BM_Conv2D(N, H, W, C, FC, FH, FW, type, LABEL) \ argument
331 #define BM_Conv2D(N, H, W, C, FC, FH, FW, type, LABEL) \ argument
335 #define BM_Conv2DBwdInputT(kind, N, H, W, C, FC, FH, FW, type, LABEL) \ argument
350 #define BM_Conv2DBwdInput(N, H, W, C, FC, FH, FW, type, LABEL) \ argument
354 #define BM_Conv2DBwdInput(N, H, W, C, FC, FH, FW, type, LABEL) \ argument
358 #define BM_Conv2DBwdFilterT(kind, N, H, W, C, FC, FH, FW, type, LABEL) \ argument
373 #define BM_Conv2DBwdFilter(N, H, W, C, FC, FH, FW, type, LABEL) \ argument
377 #define BM_Conv2DBwdFilter(N, H, W, C, FC, FH, FW, type, LABEL) \ argument
|
| D | conv_ops_benchmark_test.cc | 316 #define BM_NAME(name, type, N, H, W, C, FW, FH, FC) \ argument
319 #define BM_Conv2D(N, H, W, C, FW, FH, FC, type, LABEL) \ argument
327 #define BM_Conv2DWithBias(N, H, W, C, FW, FH, FC, type, LABEL) \ argument
337 #define BM_Conv2DWithBiasAndRelu(N, H, W, C, FW, FH, FC, type, LABEL) \ argument
348 #define BM_FusedConv2DWithBias(N, H, W, C, FW, FH, FC, type, LABEL) \ argument
358 #define BM_FusedConv2DWithBiasAndRelu(N, H, W, C, FW, FH, FC, type, LABEL) \ argument
369 #define BM_Conv2DWithBatchNorm(N, H, W, C, FW, FH, FC, type, LABEL) \ argument
379 #define BM_Conv2DWithBatchNormAndRelu(N, H, W, C, FW, FH, FC, type, LABEL) \ argument
391 #define BM_FusedConv2DWithBatchNorm(N, H, W, C, FW, FH, FC, type, LABEL) \ argument
401 #define BM_FusedConv2DWithBatchNormAndRelu(N, H, W, C, FW, FH, FC, type, \ argument
[all …]
|
| D | conv_grad_input_ops_benchmark_test.cc | 111 #define BM_NAME(name, type, T, FMT, N, H, W, C, FH, FW, FC, SH, SW, PADDING) \ argument
115 #define BM_Conv2DBwdInput(T, FMT, N, H, W, C, FW, FH, FC, SH, SW, PADDING, \ argument
|
| D | conv_grad_filter_ops_benchmark_test.cc | 112 #define BM_NAME(name, type, T, FMT, N, H, W, C, FH, FW, FC, SH, SW, PADDING) \ argument
116 #define BM_Conv2DBwdFilter(T, FMT, N, H, W, C, FH, FW, FC, SH, SW, PADDING, \ argument
|
| D | eigen_spatial_convolutions_test.cc | 1772 #define BM_RHS_NAME(prefix, T, N, H, W, C, FC, FH, FW, PAD, SH, SW, ISH, ISW, \ argument
1778 #define BM_PackRhs(T, N, H, W, C, FC, FH, FW, PAD, SH, SW, ISH, ISW, BR, BC) \ argument
2019 #define BM_LHS_NAME(prefix, T, C, FC, FH, FW, BR, BC) \ argument
2022 #define BM_PackLhs(T, C, FC, FH, FW, BR, BC) \ argument
|