| /external/tensorflow/tensorflow/core/kernels/ |
| D | eigen_benchmark_cpu_test.cc | 120 #define BM_SPATIAL_NAME(prefix, NT, N, H, W, C, FC, FH, FW) \ argument
123 #define BM_SpatialConvolution(NT, N, H, W, C, FC, FH, FW, LABEL) \ argument
131 #define BM_SpatialConvolutionBwdInput(NT, N, H, W, C, FC, FH, FW, LABEL) \ argument
140 #define BM_SpatialConvolutionBwdKernel(NT, N, H, W, C, FC, FH, FW, LABEL) \ argument
149 #define BM_SpatialConvolutions(N, H, W, C, FC, FH, FW, LABEL) \ argument
155 #define BM_SpatialConvolutionsBwdInput(N, H, W, C, FC, FH, FW, LABEL) \ argument
161 #define BM_SpatialConvolutionsBwdKernel(N, H, W, C, FC, FH, FW, LABEL) \ argument
351 #define BM_CUBOID_NAME(p, NT, N, H, W, P, C, FC, FH, FW, FP) \ argument
355 #define BM_CuboidConvolution(NT, N, H, W, P, C, FC, FH, FW, FP, LABEL) \ argument
363 #define BM_CuboidConvolutionBwdInput(NT, N, H, W, P, C, FC, FH, FW, FP, LABEL) \ argument
[all …]
|
| D | conv_ops_benchmark_test.cc | 336 #define BM_NAME(name, type, N, H, W, C, FW, FH, FC) \ argument
339 #define BM_Conv2D(N, H, W, C, FW, FH, FC, type, LABEL) \ argument
350 #define BM_Conv2DWithBias(N, H, W, C, FW, FH, FC, type, LABEL) \ argument
362 #define BM_Conv2DWithBiasAndRelu(N, H, W, C, FW, FH, FC, type, LABEL) \ argument
376 #define BM_FusedConv2DWithBias(N, H, W, C, FW, FH, FC, type, LABEL) \ argument
389 #define BM_FusedConv2DWithBiasAndRelu(N, H, W, C, FW, FH, FC, type, LABEL) \ argument
403 #define BM_Conv2DWithBatchNorm(N, H, W, C, FW, FH, FC, type, LABEL) \ argument
415 #define BM_Conv2DWithBatchNormAndRelu(N, H, W, C, FW, FH, FC, type, LABEL) \ argument
430 #define BM_FusedConv2DWithBatchNorm(N, H, W, C, FW, FH, FC, type, LABEL) \ argument
444 #define BM_FusedConv2DWithBatchNormAndRelu(N, H, W, C, FW, FH, FC, type, \ argument
[all …]
|
| 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 | 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 | eigen_spatial_convolutions_test.cc | 1763 #define BM_RHS_NAME(prefix, T, N, H, W, C, FC, FH, FW, PAD, SH, SW, ISH, ISW, \ argument
1769 #define BM_PackRhs(T, N, H, W, C, FC, FH, FW, PAD, SH, SW, ISH, ISW, BR, BC) \ argument
2012 #define BM_LHS_NAME(prefix, T, C, FC, FH, FW, BR, BC) \ argument
2015 #define BM_PackLhs(T, C, FC, FH, FW, BR, BC) \ argument
|
| /external/tensorflow/tensorflow/core/kernels/mkl/ |
| D | mkl_conv_ops_test.cc | 285 #define BM_NAME(p, type, N, H, W, C, FC, FH, FW) \ argument
291 #define BM_Conv2DT(kind, N, H, W, C, FC, FH, FW, type, LABEL) \ argument
307 #define BM_Conv2D(N, H, W, C, FC, FH, FW, type, LABEL) \ argument
311 #define BM_Conv2DBwdInputT(kind, N, H, W, C, FC, FH, FW, type, LABEL) \ argument
327 #define BM_Conv2DBwdInput(N, H, W, C, FC, FH, FW, type, LABEL) \ argument
331 #define BM_Conv2DBwdFilterT(kind, N, H, W, C, FC, FH, FW, type, LABEL) \ argument
347 #define BM_Conv2DBwdFilter(N, H, W, C, FC, FH, FW, type, LABEL) \ argument
|