Searched refs:kernel_layout (Results 1 – 5 of 5) sorted by relevance
| /external/tensorflow/tensorflow/compiler/xla/tests/ |
| D | conv_depthwise_common.cc | 43 absl::StrJoin(spec.kernel_layout, "_"), "_output_dims_", in DepthwiseConvolution2DTestDataToString()
76 absl::StrJoin(spec.kernel_layout, ","), data_type, in BuildHloTextDepthwiseConvolution2D()
82 absl::StrJoin(spec.kernel_layout, ","), spec.window, spec.window, in BuildHloTextDepthwiseConvolution2D()
100 absl::StrJoin(spec.kernel_layout, ","), data_type, in BuildHloTextDepthwiseConvolution2D()
106 absl::StrJoin(spec.kernel_layout, ","), spec.window, spec.window, in BuildHloTextDepthwiseConvolution2D()
124 absl::StrJoin(spec.kernel_layout, ","), data_type, in BuildHloTextDepthwiseConvolution2D()
130 absl::StrJoin(spec.kernel_layout, ","), spec.window, spec.window, in BuildHloTextDepthwiseConvolution2D()
|
| D | grouped_convolution_test.cc | 45 std::vector<int64> kernel_layout; member
73 std::vector<int64> kernel_layout = {3, 2, 1, 0}; in GetConv2DTestCases() local
85 config.kernel_layout = {3, 2, 1, 0}; in GetConv2DTestCases()
134 absl::StrJoin(spec.kernel_layout, "_"), "_output_dims_", in GroupedConvolution2DTestDataToString()
167 absl::StrJoin(spec.kernel_layout, ","), data_type, in BuildHloTextGroupedConvolution2D()
173 absl::StrJoin(spec.kernel_layout, ","), spec.window, spec.window, in BuildHloTextGroupedConvolution2D()
193 absl::StrJoin(spec.kernel_layout, ","), data_type, in BuildHloTextGroupedConvolution2D()
199 absl::StrJoin(spec.kernel_layout, ","), spec.window, spec.window, in BuildHloTextGroupedConvolution2D()
218 absl::StrJoin(spec.kernel_layout, ","), data_type, in BuildHloTextGroupedConvolution2D()
224 absl::StrJoin(spec.kernel_layout, ","), spec.window, spec.window, in BuildHloTextGroupedConvolution2D()
|
| D | conv_depthwise_test.cc | 50 std::vector<int64> kernel_layout = {3, 2, 1, 0}; in GetConv2DTestCases() local
59 config.kernel_layout = {3, 2, 1, 0}; in GetConv2DTestCases()
|
| D | conv_depthwise_common.h | 39 std::vector<int64> kernel_layout; member
|
| /external/ruy/ruy/ |
| D | create_trmul_params.h | 43 const KernelLayout& kernel_layout, in CreatePackedLayout() argument
48 packed_layout->rows = round_up_pot(src.rows, kernel_layout.rows); in CreatePackedLayout()
49 packed_layout->cols = round_up_pot(src.cols, kernel_layout.cols); in CreatePackedLayout()
51 packed_layout->kernel = kernel_layout; in CreatePackedLayout()
55 void CreatePackedMatrix(Side side, const KernelLayout& kernel_layout, in CreatePackedMatrix() argument
70 CreatePackedLayout(src.layout, kernel_layout, &packed_matrix->layout); in CreatePackedMatrix()
|