Searched defs:kernel_cols (Results 1 – 11 of 11) sorted by relevance
| /external/tensorflow/tensorflow/compiler/xla/service/cpu/ |
| D | runtime_single_threaded_conv2d.cc | 28 tensorflow::int64 kernel_cols, tensorflow::int64 kernel_channels, in __xla_cpu_runtime_EigenSingleThreadedConvF16()
49 tensorflow::int64 kernel_rows, tensorflow::int64 kernel_cols, in __xla_cpu_runtime_EigenSingleThreadedConvF32()
|
| D | runtime_conv2d.cc | 30 tensorflow::int64 kernel_rows, tensorflow::int64 kernel_cols, in __xla_cpu_runtime_EigenConvF32()
54 tensorflow::int64 kernel_cols, tensorflow::int64 kernel_channels, in __xla_cpu_runtime_EigenConvF16()
|
| D | runtime_conv2d_mkl.cc | 55 int64 kernel_cols, int64 kernel_channels, int64 kernel_filters, in MKLConvImpl()
157 int64 kernel_rows, int64 kernel_cols, int64 kernel_channels, in __xla_cpu_runtime_MKLConvF32()
|
| D | runtime_conv2d_impl.h | 36 Eigen::Index kernel_cols, Eigen::Index kernel_channels, in EigenConvImpl()
|
| D | ir_emitter.cc | 916 int64 kernel_cols = in HandleConvolution() local
|
| /external/ruy/ruy/ |
| D | block_map_test.cc | 36 int kernel_cols, int lhs_scalar_size, in MakeBlockMapTuningTest()
|
| D | block_map.cc | 171 void GetRectangularness(int rows, int cols, int kernel_rows, int kernel_cols, in GetRectangularness()
339 int kernel_cols, int lhs_scalar_size, int rhs_scalar_size, in MakeBlockMap()
|
| D | pack_arm.cc | 2257 int input_xor, int kernel_cols) { in Pack8bitRowMajorForNeon()
|
| /external/tensorflow/tensorflow/core/kernels/ |
| D | nn_ops_test.cc | 839 int kernel_cols, int stride, Padding padding, in BM_AvgPool()
929 int kernel_cols, int stride, Padding padding, in BM_AvgPoolBk()
1031 int kernel_cols, int stride, Padding padding, in BM_MaxPool()
1125 int kernel_cols, int stride, Padding padding, in BM_MaxPoolBk()
|
| /external/tensorflow/tensorflow/core/framework/ |
| D | common_shape_fns.cc | 1090 int32 kernel_cols = GetTensorDim(kernel_sizes, data_format, 'W'); in AvgPoolShape() local
1533 int32 kernel_cols = GetTensorDim(kernel_sizes, data_format, 'W'); in MaxPoolShapeImpl() local
1672 int32 kernel_cols = GetTensorDim(kernel_sizes, data_format, 'W'); in MaxPoolV2Shape() local
1729 int32 kernel_planes, kernel_rows, kernel_cols; in Pool3DShape() local
|
| /external/tensorflow/tensorflow/compiler/tests/ |
| D | randomized_tests.cc | 1113 int kernel_cols = in TEST_F() local
2317 int kernel_cols = in TEST_F() local
|