Home
last modified time | relevance | path

Searched refs:kernel_layout (Results 1 – 2 of 2) sorted by relevance

/external/tensorflow/tensorflow/compiler/xla/tests/
Dconv_depthwise_test.cc40 std::vector<int64> kernel_layout; member
63 std::vector<int64> kernel_layout = {3, 2, 1, 0}; in GetConv2DTestCases() local
72 config.kernel_layout = {3, 2, 1, 0}; in GetConv2DTestCases()
120 absl::StrJoin(spec.kernel_layout, "_"), "_output_dims_", in DepthwiseConvolution2DTestDataToString()
152 absl::StrJoin(spec.kernel_layout, ","), data_type, in BuildHloTextDepthwiseConvolution2D()
158 absl::StrJoin(spec.kernel_layout, ","), spec.window, spec.window, in BuildHloTextDepthwiseConvolution2D()
177 absl::StrJoin(spec.kernel_layout, ","), data_type, in BuildHloTextDepthwiseConvolution2D()
183 absl::StrJoin(spec.kernel_layout, ","), spec.window, spec.window, in BuildHloTextDepthwiseConvolution2D()
201 absl::StrJoin(spec.kernel_layout, ","), data_type, in BuildHloTextDepthwiseConvolution2D()
207 absl::StrJoin(spec.kernel_layout, ","), spec.window, spec.window, in BuildHloTextDepthwiseConvolution2D()
Dgrouped_convolution_test.cc41 std::vector<int64> kernel_layout; member
69 std::vector<int64> kernel_layout = {3, 2, 1, 0}; in GetConv2DTestCases() local
81 config.kernel_layout = {3, 2, 1, 0}; in GetConv2DTestCases()
129 absl::StrJoin(spec.kernel_layout, "_"), "_output_dims_", in GroupedConvolution2DTestDataToString()
162 absl::StrJoin(spec.kernel_layout, ","), data_type, in BuildHloTextGroupedConvolution2D()
168 absl::StrJoin(spec.kernel_layout, ","), spec.window, spec.window, in BuildHloTextGroupedConvolution2D()
188 absl::StrJoin(spec.kernel_layout, ","), data_type, in BuildHloTextGroupedConvolution2D()
194 absl::StrJoin(spec.kernel_layout, ","), spec.window, spec.window, in BuildHloTextGroupedConvolution2D()
213 absl::StrJoin(spec.kernel_layout, ","), data_type, in BuildHloTextGroupedConvolution2D()
219 absl::StrJoin(spec.kernel_layout, ","), spec.window, spec.window, in BuildHloTextGroupedConvolution2D()