| /external/ruy/ruy/ |
| D | block_map.cc | 171 void GetRectangularness(int rows, int cols, int kernel_rows, int kernel_cols, in GetRectangularness() argument
186 ceil_log2(cols) - pot_log2(kernel_cols); in GetRectangularness()
204 std::max(0, floor_log2(cols) - pot_log2(kernel_cols) - in GetRectangularness()
339 int kernel_cols, int lhs_scalar_size, int rhs_scalar_size, in MakeBlockMap() argument
346 RUY_DCHECK_GE(cols, kernel_cols); in MakeBlockMap()
348 RUY_DCHECK_EQ(cols % kernel_cols, 0); in MakeBlockMap()
354 GetRectangularness(rows, cols, kernel_rows, kernel_cols, in MakeBlockMap()
358 const int kernel_cols_log2 = pot_log2(kernel_cols); in MakeBlockMap()
439 round_down_pot(cols >> num_blocks_of_cols_log2, kernel_cols); in MakeBlockMap()
446 kernel_cols) >> in MakeBlockMap()
[all …]
|
| D | pack_arm.cc | 2257 int input_xor, int kernel_cols) { in Pack8bitRowMajorForNeon() argument
2432 dst_ptr += (kernel_cols == 2) ? 2 * packed_stride : 32; in Pack8bitRowMajorForNeon()
2439 dst_ptr += (kernel_cols == 2) ? 2 * packed_stride : 32; in Pack8bitRowMajorForNeon()
2449 std::int8_t* dst_ptr = packed_ptr + (col & (kernel_cols - 1)) * 16; in Pack8bitRowMajorForNeon()
2461 if (((col + 1) & (kernel_cols - 1)) == 0) { in Pack8bitRowMajorForNeon()
2462 packed_ptr += kernel_cols * packed_stride; in Pack8bitRowMajorForNeon()
2470 std::int8_t* dst_ptr = packed_ptr + (col & (kernel_cols - 1)) * 16; in Pack8bitRowMajorForNeon()
2472 if (((col + 1) & (kernel_cols - 1)) == 0) { in Pack8bitRowMajorForNeon()
2473 packed_ptr += kernel_cols * packed_stride; in Pack8bitRowMajorForNeon()
|
| D | block_map_test.cc | 36 int kernel_cols, int lhs_scalar_size, in MakeBlockMapTuningTest() argument
45 MakeBlockMap(rows, cols, depth, kernel_rows, kernel_cols, lhs_scalar_size, in MakeBlockMapTuningTest()
|
| D | block_map.h | 119 int kernel_cols, int lhs_scalar_size, int rhs_scalar_size,
|
| D | trace.h | 626 ThreadLocalTrace()->Write("Kernel block: %dx%d", kernel_rows, kernel_cols); \
|
| D | pack_arm.h | 60 int input_xor, int kernel_cols);
|
| /external/tensorflow/tensorflow/compiler/xla/service/cpu/ |
| D | runtime_conv2d_impl.h | 36 Eigen::Index kernel_cols, Eigen::Index kernel_channels, in EigenConvImpl() argument
50 kernel(rhs, kernel_rows, kernel_cols, kernel_channels, kernel_filters); in EigenConvImpl()
65 pre_contract_dims[1] = kernel_channels * kernel_cols * kernel_rows; in EigenConvImpl()
75 kernel_dims[0] = kernel_channels * kernel_cols * kernel_rows; in EigenConvImpl()
81 .extract_image_patches(kernel_cols, kernel_rows, col_stride, in EigenConvImpl()
|
| D | runtime_single_threaded_conv2d.cc | 28 tensorflow::int64 kernel_cols, tensorflow::int64 kernel_channels, in __xla_cpu_runtime_EigenSingleThreadedConvF16() argument
38 input_cols, input_channels, kernel_rows, kernel_cols, kernel_channels, in __xla_cpu_runtime_EigenSingleThreadedConvF16()
49 tensorflow::int64 kernel_rows, tensorflow::int64 kernel_cols, in __xla_cpu_runtime_EigenSingleThreadedConvF32() argument
59 input_cols, input_channels, kernel_rows, kernel_cols, kernel_channels, in __xla_cpu_runtime_EigenSingleThreadedConvF32()
|
| D | runtime_conv2d.cc | 30 tensorflow::int64 kernel_rows, tensorflow::int64 kernel_cols, in __xla_cpu_runtime_EigenConvF32() argument
43 input_rows, input_cols, input_channels, kernel_rows, kernel_cols, in __xla_cpu_runtime_EigenConvF32()
54 tensorflow::int64 kernel_cols, tensorflow::int64 kernel_channels, in __xla_cpu_runtime_EigenConvF16() argument
67 input_rows, input_cols, input_channels, kernel_rows, kernel_cols, in __xla_cpu_runtime_EigenConvF16()
|
| D | runtime_conv2d_mkl.cc | 55 int64 kernel_cols, int64 kernel_channels, int64 kernel_filters, in MKLConvImpl() argument
72 ToInt(kernel_cols)}; in MKLConvImpl()
157 int64 kernel_rows, int64 kernel_cols, int64 kernel_channels, in __xla_cpu_runtime_MKLConvF32() argument
168 input_channels, kernel_rows, kernel_cols, kernel_channels, in __xla_cpu_runtime_MKLConvF32()
174 input_channels, kernel_rows, kernel_cols, kernel_channels, in __xla_cpu_runtime_MKLConvF32()
|
| D | runtime_conv2d.h | 29 tensorflow::int64 kernel_cols, tensorflow::int64 kernel_channels,
43 tensorflow::int64 kernel_rows, tensorflow::int64 kernel_cols,
|
| D | runtime_single_threaded_conv2d.h | 29 tensorflow::int64 kernel_rows, tensorflow::int64 kernel_cols,
43 tensorflow::int64 kernel_cols, tensorflow::int64 kernel_channels,
|
| D | runtime_conv2d_mkl.h | 29 tensorflow::int64 kernel_cols, tensorflow::int64 kernel_channels,
|
| D | ir_emitter.cc | 916 int64 kernel_cols = in HandleConvolution() local
989 b_.getInt64(kernel_cols), in HandleConvolution()
|
| /external/tensorflow/tensorflow/core/kernels/ |
| D | nn_ops_test.cc | 839 int kernel_cols, int stride, Padding padding, in BM_AvgPool() argument
857 CHECK_EQ(kernel_rows, kernel_cols); in BM_AvgPool()
860 .Attr("ksize", {1, kernel_rows, kernel_cols, 1}) in BM_AvgPool()
929 int kernel_cols, int stride, Padding padding, in BM_AvgPoolBk() argument
944 TF_CHECK_OK(GetWindowedOutputSize(cols, kernel_cols, stride, padding, in BM_AvgPoolBk()
964 .Attr("ksize", {1, kernel_rows, kernel_cols, 1}) in BM_AvgPoolBk()
1031 int kernel_cols, int stride, Padding padding, in BM_MaxPool() argument
1052 CHECK_EQ(kernel_rows, kernel_cols); in BM_MaxPool()
1055 .Attr("ksize", {1, kernel_rows, kernel_cols, 1}) in BM_MaxPool()
1125 int kernel_cols, int stride, Padding padding, in BM_MaxPoolBk() argument
[all …]
|
| /external/tensorflow/tensorflow/core/framework/ |
| D | common_shape_fns.cc | 1090 int32 kernel_cols = GetTensorDim(kernel_sizes, data_format, 'W'); in AvgPoolShape() local
1113 c, in_cols_dim, kernel_cols, stride_cols, padding, &output_cols)); in AvgPoolShape()
1533 int32 kernel_cols = GetTensorDim(kernel_sizes, data_format, 'W'); in MaxPoolShapeImpl() local
1576 c, in_cols_dim, kernel_cols, /*dilation_rate=*/1, stride_cols, padding, in MaxPoolShapeImpl()
1672 int32 kernel_cols = GetTensorDim(kernel_sizes, data_format, 'W'); in MaxPoolV2Shape() local
1692 c, in_cols_dim, kernel_cols, stride_cols, padding, &output_cols)); in MaxPoolV2Shape()
1729 int32 kernel_planes, kernel_rows, kernel_cols; in Pool3DShape() local
1743 kernel_cols = kernel_sizes[4]; in Pool3DShape()
1750 kernel_cols = kernel_sizes[3]; in Pool3DShape()
1771 c, in_cols_dim, kernel_cols, stride_cols, padding, &output_cols)); in Pool3DShape()
|
| /external/tensorflow/tensorflow/compiler/tests/ |
| D | randomized_tests.cc | 1113 int kernel_cols = in TEST_F() local
1122 .Attr("ksize", {1, kernel_rows, kernel_cols, 1}) in TEST_F()
2317 int kernel_cols = in TEST_F() local
2327 .Attr("ksize", {1, kernel_rows, kernel_cols, 1}) in TEST_F()
|