| /external/tensorflow/tensorflow/compiler/xla/tests/ |
| reduce_hlo_test.cc | 91 LayoutUtil::MakeLayout(reduce_layout.output_minor_to_major);
96 LayoutUtil::MakeLayout(reduce_layout.input_minor_to_major);
|
| dot_operation_test.cc | 180 LayoutUtil::MakeLayout(MinorToMajorForIsRowMajor(lhs_row_major))))
186 LayoutUtil::MakeLayout(MinorToMajorForIsRowMajor(rhs_row_major))))
236 *dot_lhs_data, LayoutUtil::MakeLayout(
244 *dot_rhs_data, LayoutUtil::MakeLayout(
256 *addend_data, LayoutUtil::MakeLayout(
389 LayoutUtil::MakeLayout(MinorToMajorForIsRowMajor(lhs_row_major))))
395 LayoutUtil::MakeLayout(MinorToMajorForIsRowMajor(rhs_row_major))))
450 {{1.0, 2.0, 3.0, -4.0}}, LayoutUtil::MakeLayout({1, 0})))
456 LayoutUtil::MakeLayout({1, 0})))
578 *lhs, LayoutUtil::MakeLayout(
[all...] |
| round_trip_packed_literal_test.cc | 87 const Layout layout = LayoutUtil::MakeLayout({1, 0});
122 const Layout layout = LayoutUtil::MakeLayout({0, 1});
|
| matrix_ops_simple_test.cc | 184 lhs, LayoutUtil::MakeLayout(minor_to_major(row_major)))));
188 rhs, LayoutUtil::MakeLayout(minor_to_major(row_major)))));
251 lhs, LayoutUtil::MakeLayout(minor_to_major(row_major)))));
256 rhs, LayoutUtil::MakeLayout(minor_to_major(row_major)))));
|
| reshape_test.cc | 635 LayoutUtil::MakeLayout({0, 1, 2, 3}));
[all...] |
| compilation_cache_test.cc | 142 {{1.0f, 2.0f}, {3.0f, 4.0f}}, LayoutUtil::MakeLayout({1, 0}));
147 {{1.0f, 2.0f}, {3.0f, 4.0f}}, LayoutUtil::MakeLayout({0, 1}));
|
| slice_test.cc | 174 *expected, LayoutUtil::MakeLayout({0, 1, 2, 3}));
315 input, LayoutUtil::MakeLayout(spec.layout));
405 values, LayoutUtil::MakeLayout(spec.input_layout));
|
| client_test.cc | 59 {{11, 22}, {33, 44}}, LayoutUtil::MakeLayout(transfer_layout));
|
| compute_constant_test.cc | 258 auto layout_proto = LayoutUtil::MakeLayout(layout);
268 LayoutUtil::MakeLayout(layout));
|
| copy_test.cc | 182 ForceResultLayout(module.get(), LayoutUtil::MakeLayout({1, 2, 0}));
216 ForceResultLayout(module.get(), LayoutUtil::MakeLayout(permutation));
|
| reduce_test.cc | 175 input_literal->Relayout(LayoutUtil::MakeLayout({minor, major}));
209 input_literal->Relayout(LayoutUtil::MakeLayout({minor, major}));
237 input_literal->Relayout(LayoutUtil::MakeLayout({minor, major}));
295 input_literal->Relayout(LayoutUtil::MakeLayout({minor, major}));
449 input_literal = input_literal->Relayout(LayoutUtil::MakeLayout({0, 1}));
481 input_literal = input_literal->Relayout(LayoutUtil::MakeLayout({0, 1}));
[all...] |
| reduce_window_test.cc | 298 input_array, LayoutUtil::MakeLayout({0, 3, 2, 1}));
366 input_array, LayoutUtil::MakeLayout({3, 2, 1, 0}));
388 input_array, LayoutUtil::MakeLayout({3, 2, 1, 0}));
410 input_array, LayoutUtil::MakeLayout({3, 2, 1, 0}));
581 input, LayoutUtil::MakeLayout(param.layout));
[all...] |
| params_test.cc | 466 {{1, 2}, {3, 4}}, LayoutUtil::MakeLayout({0, 1}));
478 {{1, 3}, {2, 4}}, LayoutUtil::MakeLayout({1, 0}));
500 LayoutUtil::MakeLayout(original_layout);
|
| local_client_execute_test.cc | 140 {{1.0f, 2.0f}, {3.0f, 4.0f}}, LayoutUtil::MakeLayout({0, 1})));
142 LayoutUtil::MakeLayout({0, 1})));
146 {{10.0f, 20.0f}, {30.0f, 40.0f}}, LayoutUtil::MakeLayout({1, 0})));
148 LayoutUtil::MakeLayout({1, 0})));
183 LayoutUtil::MakeLayout({0, 1})));
195 LayoutUtil::MakeLayout({1, 0})));
[all...] |
| /external/tensorflow/tensorflow/compiler/xla/service/ |
| layout_assignment_test.cc | 77 Layout layout = LayoutUtil::MakeLayout(minor_to_major);
107 Layout col_major_layout = LayoutUtil::MakeLayout({1, 0});
112 Layout row_major_layout = LayoutUtil::MakeLayout({0, 1});
137 {{1.0, 2.0}, {3.0, 4.0}}, LayoutUtil::MakeLayout(minor_to_major));
139 {{5.0, 6.0}, {7.0, 8.0}}, LayoutUtil::MakeLayout(minor_to_major));
159 Layout layout = LayoutUtil::MakeLayout(minor_to_major);
188 {{1.0, 2.0}, {3.0, 4.0}}, LayoutUtil::MakeLayout({0, 1}))));
191 {{1.0, 2.0}, {3.0, 4.0}}, LayoutUtil::MakeLayout({1, 0}))));
225 {{1.0, 2.0}, {3.0, 4.0}}, LayoutUtil::MakeLayout({0, 1}))));
228 {{1.0, 2.0}, {3.0, 4.0}}, LayoutUtil::MakeLayout({1, 0}))))
[all...] |
| hlo_cse_test.cc | 84 {{1.0, 2.0}, {3.0, 4.0}}, LayoutUtil::MakeLayout({0, 1}))));
87 {{1.0, 2.0}, {3.0, 4.0}}, LayoutUtil::MakeLayout({1, 0}))));
116 {{1.0, 2.0}, {3.0, 4.0}}, LayoutUtil::MakeLayout({0, 1}))));
119 {{1.0, 2.0}, {3.0, 4.0}}, LayoutUtil::MakeLayout({1, 0}))));
244 *exp1->mutable_shape()->mutable_layout() = LayoutUtil::MakeLayout({0, 1});
248 *exp2->mutable_shape()->mutable_layout() = LayoutUtil::MakeLayout({1, 0});
275 *exp1->mutable_shape()->mutable_layout() = LayoutUtil::MakeLayout({0, 1});
279 *exp2->mutable_shape()->mutable_layout() = LayoutUtil::MakeLayout({1, 0});
|
| algebraic_simplifier_test.cc | [all...] |
| /external/tensorflow/tensorflow/compiler/xla/service/cpu/ |
| cpu_layout_assignment_test.cc | 79 EXPECT_TRUE(LayoutUtil::Equal(LayoutUtil::MakeLayout({1, 0}),
81 EXPECT_TRUE(LayoutUtil::Equal(LayoutUtil::MakeLayout({0, 1}),
83 EXPECT_TRUE(LayoutUtil::Equal(LayoutUtil::MakeLayout({1, 0}),
120 EXPECT_TRUE(LayoutUtil::Equal(LayoutUtil::MakeLayout({0, 1}),
124 EXPECT_TRUE(LayoutUtil::Equal(LayoutUtil::MakeLayout({1, 0}),
168 EXPECT_TRUE(LayoutUtil::Equal(LayoutUtil::MakeLayout({1, 0}),
199 EXPECT_TRUE(LayoutUtil::Equal(LayoutUtil::MakeLayout({1, 0}),
239 EXPECT_TRUE(LayoutUtil::Equal(LayoutUtil::MakeLayout({1, 0}),
326 ? LayoutUtil::MakeLayout({0, 1})
327 : LayoutUtil::MakeLayout({1, 0})
[all...] |
| cpu_layout_assignment.cc | 84 *new_shape.mutable_layout() = LayoutUtil::MakeLayout(dimension_order);
92 *new_shape.mutable_layout() = LayoutUtil::MakeLayout(dimension_order);
|
| external_constant_pool_test.cc | 52 {{1, 2}, {3, 4}}, LayoutUtil::MakeLayout({0, 1}));
|
| /external/tensorflow/tensorflow/compiler/xla/ |
| layout_util_test.cc | 33 *shape.mutable_layout() = LayoutUtil::MakeLayout(minor_to_major);
211 *src.mutable_layout() = LayoutUtil::MakeLayout({1, 2, 3, 4});
281 EXPECT_TRUE(LayoutUtil::Equal(LayoutUtil::MakeLayout({1, 0}),
283 EXPECT_TRUE(LayoutUtil::Equal(LayoutUtil::MakeLayout({2, 1, 0}),
285 EXPECT_TRUE(LayoutUtil::Equal(LayoutUtil::MakeLayout({3, 2, 1, 0}),
288 LayoutUtil::Equal(LayoutUtil::MakeLayout({4, 3, 2, 1, 0}),
300 oss << LayoutUtil::MakeLayout({0, 1, 2});
|
| layout_util.h | 37 static Layout MakeLayout(tensorflow::gtl::ArraySlice<int64> minor_to_major);
|
| shape_util_test.cc | 625 Layout input_layout = input_is_row_major ? LayoutUtil::MakeLayout({1, 0})
626 : LayoutUtil::MakeLayout({0, 1});
628 ? LayoutUtil::MakeLayout({1, 0})
629 : LayoutUtil::MakeLayout({0, 1});
|
| /external/tensorflow/tensorflow/compiler/xla/service/gpu/ |
| gpu_layout_assignment.cc | 83 *input_shape.mutable_layout() = LayoutUtil::MakeLayout(input_layout);
92 *filter_shape.mutable_layout() = LayoutUtil::MakeLayout(filter_layout);
101 *output_shape.mutable_layout() = LayoutUtil::MakeLayout(output_layout);
|
| gpu_layout_assignment_test.cc | 40 *ashape_in_row_major.mutable_layout() = LayoutUtil::MakeLayout({1, 0});
41 *ashape_in_col_major.mutable_layout() = LayoutUtil::MakeLayout({0, 1});
96 *shapes.back().mutable_layout() = LayoutUtil::MakeLayout(layout_vec);
|
