Home
last modified time | relevance | path

Searched refs:MakeLayout (Results 1 – 25 of 48) sorted by relevance

12

/external/tensorflow/tensorflow/compiler/xla/service/cpu/
Dcpu_layout_assignment_test.cc86 EXPECT_TRUE(LayoutUtil::Equal(LayoutUtil::MakeLayout({1, 0}), in TEST_F()
88 EXPECT_TRUE(LayoutUtil::Equal(LayoutUtil::MakeLayout({0, 1}), in TEST_F()
90 EXPECT_TRUE(LayoutUtil::Equal(LayoutUtil::MakeLayout({1, 0}), in TEST_F()
127 EXPECT_TRUE(LayoutUtil::Equal(LayoutUtil::MakeLayout({0, 1}), in TEST_F()
131 EXPECT_TRUE(LayoutUtil::Equal(LayoutUtil::MakeLayout({1, 0}), in TEST_F()
175 EXPECT_TRUE(LayoutUtil::Equal(LayoutUtil::MakeLayout({1, 0}), in TEST_F()
206 EXPECT_TRUE(LayoutUtil::Equal(LayoutUtil::MakeLayout({1, 0}), in TEST_F()
246 EXPECT_TRUE(LayoutUtil::Equal(LayoutUtil::MakeLayout({1, 0}), in TEST_F()
339 ? LayoutUtil::MakeLayout({0, 1}) in AssertCorrectLayoutForDotOutputFusion()
340 : LayoutUtil::MakeLayout({1, 0}); in AssertCorrectLayoutForDotOutputFusion()
[all …]
Dcpu_layout_assignment.cc84 *new_shape.mutable_layout() = LayoutUtil::MakeLayout(dimension_order); in RowMajorShape()
92 *new_shape.mutable_layout() = LayoutUtil::MakeLayout(dimension_order); in ColMajorShape()
/external/tensorflow/tensorflow/compiler/xla/service/
Dlayout_assignment_test.cc70 const Layout expected = LayoutUtil::MakeLayout(minor_to_major); in ExpectLayoutIs()
80 const Layout expected = LayoutUtil::MakeLayout(minor_to_major); in ExpectTupleLayoutIs()
106 Layout layout = LayoutUtil::MakeLayout(minor_to_major); in TEST_F()
136 Layout col_major_layout = LayoutUtil::MakeLayout({1, 0}); in TEST_F()
141 Layout row_major_layout = LayoutUtil::MakeLayout({0, 1}); in TEST_F()
166 {{1.0, 2.0}, {3.0, 4.0}}, LayoutUtil::MakeLayout(minor_to_major)); in TEST_F()
168 {{5.0, 6.0}, {7.0, 8.0}}, LayoutUtil::MakeLayout(minor_to_major)); in TEST_F()
188 Layout layout = LayoutUtil::MakeLayout(minor_to_major); in TEST_F()
217 {{1.0, 2.0}, {3.0, 4.0}}, LayoutUtil::MakeLayout({0, 1})))); in TEST_F()
220 {{1.0, 2.0}, {3.0, 4.0}}, LayoutUtil::MakeLayout({1, 0})))); in TEST_F()
[all …]
Dhlo_cse_test.cc85 {{1.0, 2.0}, {3.0, 4.0}}, LayoutUtil::MakeLayout({0, 1})))); in TEST_F()
88 {{1.0, 2.0}, {3.0, 4.0}}, LayoutUtil::MakeLayout({1, 0})))); in TEST_F()
117 {{1.0, 2.0}, {3.0, 4.0}}, LayoutUtil::MakeLayout({0, 1})))); in TEST_F()
120 {{1.0, 2.0}, {3.0, 4.0}}, LayoutUtil::MakeLayout({1, 0})))); in TEST_F()
433 *exp1->mutable_shape()->mutable_layout() = LayoutUtil::MakeLayout({0, 1}); in TEST_F()
437 *exp2->mutable_shape()->mutable_layout() = LayoutUtil::MakeLayout({1, 0}); in TEST_F()
464 *exp1->mutable_shape()->mutable_layout() = LayoutUtil::MakeLayout({0, 1}); in TEST_F()
468 *exp2->mutable_shape()->mutable_layout() = LayoutUtil::MakeLayout({1, 0}); in TEST_F()
Dpattern_matcher_gmock_test.cc57 Layout l = LayoutUtil::MakeLayout({0, 1}); in TEST()
/external/tensorflow/tensorflow/compiler/xla/tests/
Dreshape_test.cc604 LayoutUtil::MakeLayout({0, 1, 2, 3})); in XLA_TEST_P()
695 input, LayoutUtil::MakeLayout({3, 2, 1, 0})); in XLA_TEST_P()
714 input, LayoutUtil::MakeLayout({3, 2, 1, 0})); in XLA_TEST_P()
734 input, LayoutUtil::MakeLayout({3, 2, 1, 0})); in XLA_TEST_P()
760 input_array, LayoutUtil::MakeLayout({1, 2, 3, 0})); in XLA_TEST_P()
840 input, LayoutUtil::MakeLayout({3, 2, 1, 0})); in XLA_TEST_P()
850 .Relayout(LayoutUtil::MakeLayout({3, 2, 1, 0})); in XLA_TEST_P()
867 input, LayoutUtil::MakeLayout({3, 2, 1, 0})); in XLA_TEST_P()
877 .Relayout(LayoutUtil::MakeLayout({3, 2, 1, 0})); in XLA_TEST_P()
894 input, LayoutUtil::MakeLayout({3, 2, 1, 0})); in XLA_TEST_P()
[all …]
Dcustom_call_test.cc151 ForceParameterLayout(module.get(), 0, LayoutUtil::MakeLayout({1, 0})); in XLA_TEST_F()
152 ForceResultLayout(module.get(), LayoutUtil::MakeLayout({0, 1})); in XLA_TEST_F()
181 ForceParameterLayout(module.get(), 0, LayoutUtil::MakeLayout({1, 0})); in XLA_TEST_F()
182 ForceResultLayout(module.get(), LayoutUtil::MakeLayout({0, 1})); in XLA_TEST_F()
Dreduce_hlo_test.cc88 LayoutUtil::MakeLayout(reduce_layout.output_minor_to_major); in XLA_TEST_P()
93 LayoutUtil::MakeLayout(reduce_layout.input_minor_to_major); in XLA_TEST_P()
Ddot_operation_test.cc225 LayoutUtil::MakeLayout( in TestImpl()
232 LayoutUtil::MakeLayout( in TestImpl()
291 *dot_lhs_data, LayoutUtil::MakeLayout( in TestImpl()
298 Layout rhs_layout = LayoutUtil::MakeLayout( in TestImpl()
312 *addend_data, LayoutUtil::MakeLayout( in TestImpl()
473 LayoutUtil::MakeLayout( in TestImpl()
480 LayoutUtil::MakeLayout( in TestImpl()
506 {{1.0, 2.0, 3.0, -4.0}}, LayoutUtil::MakeLayout({1, 0}))) in XLA_TEST_F()
512 LayoutUtil::MakeLayout({1, 0}))) in XLA_TEST_F()
773 *lhs, LayoutUtil::MakeLayout( in XLA_TYPED_TEST()
[all …]
Dround_trip_packed_literal_test.cc85 const Layout layout = LayoutUtil::MakeLayout({1, 0}); in TEST_F()
118 const Layout layout = LayoutUtil::MakeLayout({0, 1}); in TEST_F()
Dcompilation_cache_test.cc145 {{1.0f, 2.0f}, {3.0f, 4.0f}}, LayoutUtil::MakeLayout({1, 0})); in XLA_TEST_F()
150 {{1.0f, 2.0f}, {3.0f, 4.0f}}, LayoutUtil::MakeLayout({0, 1})); in XLA_TEST_F()
Dreduce_test.cc181 input_literal.Relayout(LayoutUtil::MakeLayout({minor, major})); in RunR2ToR1PredTest()
214 input_literal.Relayout(LayoutUtil::MakeLayout({minor, major})); in RunR2ToR0Test()
241 input_literal.Relayout(LayoutUtil::MakeLayout({minor, major})); in RunR2ToR1Test()
298 input_literal.Relayout(LayoutUtil::MakeLayout({minor, major})); in RunVectorizedReduceTestForType()
451 input_literal = input_literal.Relayout(LayoutUtil::MakeLayout({0, 1})); in XLA_TEST_F()
482 input_literal = input_literal.Relayout(LayoutUtil::MakeLayout({0, 1})); in XLA_TEST_F()
820 input_literal.Relayout(LayoutUtil::MakeLayout(GetParam().layout)); in XLA_TEST_P()
Dreduce_window_test.cc343 input_array, LayoutUtil::MakeLayout({0, 3, 2, 1})); in TEST_P()
401 input_array, LayoutUtil::MakeLayout({3, 2, 1, 0})); in XLA_TEST_P()
422 input_array, LayoutUtil::MakeLayout({3, 2, 1, 0})); in XLA_TEST_P()
443 input_array, LayoutUtil::MakeLayout({3, 2, 1, 0})); in XLA_TEST_P()
621 input, LayoutUtil::MakeLayout(param.layout)); in DoIt()
1022 input, LayoutUtil::MakeLayout(param.layout)); in TEST_P()
1164 input, LayoutUtil::MakeLayout(param.layout)); in DoIt()
Dmatrix_ops_simple_test.cc204 lhs, LayoutUtil::MakeLayout(minor_to_major(row_major))))); in TestImpl()
208 rhs, LayoutUtil::MakeLayout(minor_to_major(row_major))))); in TestImpl()
Dlocal_client_execute_test.cc132 {{1.0f, 2.0f}, {3.0f, 4.0f}}, LayoutUtil::MakeLayout({0, 1}))); in XLA_TEST_F()
134 LayoutUtil::MakeLayout({0, 1}))); in XLA_TEST_F()
138 {{10.0f, 20.0f}, {30.0f, 40.0f}}, LayoutUtil::MakeLayout({1, 0}))); in XLA_TEST_F()
140 LayoutUtil::MakeLayout({1, 0}))); in XLA_TEST_F()
175 LayoutUtil::MakeLayout({0, 1}))); in XLA_TEST_F()
187 LayoutUtil::MakeLayout({1, 0}))); in XLA_TEST_F()
Dconvolution_test.cc780 expected_r4.Relayout(LayoutUtil::MakeLayout({0, 3, 2, 1})); in RunTest()
839 input_r4.Relayout(LayoutUtil::MakeLayout({0, 3, 2, 1})); in RunTest()
908 input_r4.Relayout(LayoutUtil::MakeLayout({0, 3, 2, 1})); in RunTest()
921 expected_r4.Relayout(LayoutUtil::MakeLayout({0, 3, 2, 1})); in RunTest()
979 input_r4.Relayout(LayoutUtil::MakeLayout({0, 3, 2, 1})); in RunTest()
996 expected_r4.Relayout(LayoutUtil::MakeLayout({0, 3, 2, 1})); in RunTest()
1118 input_r4.Relayout(LayoutUtil::MakeLayout({0, 3, 2, 1})); in RunTest()
1130 expected_r4.Relayout(LayoutUtil::MakeLayout({3, 0, 2, 1})); in RunTest()
1188 input_r4.Relayout(LayoutUtil::MakeLayout({0, 3, 2, 1})); in RunTest()
1200 expected_r4.Relayout(LayoutUtil::MakeLayout({0, 3, 2, 1})); in RunTest()
[all …]
Dslice_test.cc195 *expected, LayoutUtil::MakeLayout({0, 1, 2, 3})); in XLA_TEST_F()
395 input, LayoutUtil::MakeLayout(spec.layout)); in XLA_TEST_P()
489 values, LayoutUtil::MakeLayout(spec.input_layout)); in Run()
Dcopy_test.cc182 ForceResultLayout(module.get(), LayoutUtil::MakeLayout({1, 2, 0})); in TestCopyConstantLayout021()
216 ForceResultLayout(module.get(), LayoutUtil::MakeLayout(permutation)); in TestCopyConstantLayoutR4()
Dcompute_constant_test.cc251 auto layout_proto = LayoutUtil::MakeLayout(layout); in XLA_TEST_F()
260 {{11, 22}, {33, 44}}, LayoutUtil::MakeLayout(layout)); in XLA_TEST_F()
/external/tensorflow/tensorflow/compiler/xla/
Dlayout_util_test.cc33 *shape.mutable_layout() = LayoutUtil::MakeLayout(minor_to_major); in MakeShapeWithLayout()
211 *src.mutable_layout() = LayoutUtil::MakeLayout({1, 2, 3, 4}); in TEST_F()
308 EXPECT_TRUE(LayoutUtil::Equal(LayoutUtil::MakeLayout({1, 0}), in TEST_F()
310 EXPECT_TRUE(LayoutUtil::Equal(LayoutUtil::MakeLayout({2, 1, 0}), in TEST_F()
312 EXPECT_TRUE(LayoutUtil::Equal(LayoutUtil::MakeLayout({3, 2, 1, 0}), in TEST_F()
315 LayoutUtil::Equal(LayoutUtil::MakeLayout({4, 3, 2, 1, 0}), in TEST_F()
407 *shape.mutable_layout() = LayoutUtil::MakeLayout({0, 1, 2}); in TEST_F()
437 *shape.mutable_layout() = LayoutUtil::MakeLayout({0}); in TEST_F()
472 LayoutUtil::MakeLayout({0, 2, 3}); in TEST_F()
Dlayout_util.h39 static Layout MakeLayout(absl::Span<const int64> minor_to_major,
/external/tensorflow/tensorflow/compiler/xla/service/gpu/tests/
Dinfeed_test.cc73 const Layout r3_dim0minor = LayoutUtil::MakeLayout({0, 1, 2}); in TEST_F()
74 const Layout r3_dim0major = LayoutUtil::MakeLayout({2, 1, 0}); in TEST_F()
/external/tensorflow/tensorflow/compiler/tf2xla/
Dshape_util.cc50 xla::StatusOr<bool> MakeLayout(absl::Span<const int64> minor_to_major, in MakeLayout() function
67 *layout = xla::LayoutUtil::MakeLayout(minor_to_major); in MakeLayout()
76 TF_ASSIGN_OR_RETURN(bool has_layout, MakeLayout(minor_to_major, &layout)); in AssignLayout()
/external/tensorflow/tensorflow/compiler/xla/service/gpu/
Dgpu_layout_assignment_test.cc45 *ashape_in_row_major.mutable_layout() = LayoutUtil::MakeLayout({1, 0}); in TEST_F()
46 *ashape_in_col_major.mutable_layout() = LayoutUtil::MakeLayout({0, 1}); in TEST_F()
103 *shapes.back().mutable_layout() = LayoutUtil::MakeLayout(layout_vec); in AllLayoutsOf()
/external/tensorflow/tensorflow/compiler/xla/service/cpu/tests/
Dcpu_infeed_test.cc79 const Layout r3_dim0minor = LayoutUtil::MakeLayout({0, 1, 2}); in TEST_F()
80 const Layout r3_dim0major = LayoutUtil::MakeLayout({2, 1, 0}); in TEST_F()

12