| /external/tensorflow/tensorflow/compiler/xla/tests/ |
| D | convolution_test.cc | 612 expected_r4.Relayout(LayoutUtil::MakeLayout({0, 3, 2, 1})); in RunTest()
671 input_r4.Relayout(LayoutUtil::MakeLayout({0, 3, 2, 1})); in RunTest()
740 input_r4.Relayout(LayoutUtil::MakeLayout({0, 3, 2, 1})); in RunTest()
753 expected_r4.Relayout(LayoutUtil::MakeLayout({0, 3, 2, 1})); in RunTest()
811 input_r4.Relayout(LayoutUtil::MakeLayout({0, 3, 2, 1})); in RunTest()
828 expected_r4.Relayout(LayoutUtil::MakeLayout({0, 3, 2, 1})); in RunTest()
950 input_r4.Relayout(LayoutUtil::MakeLayout({0, 3, 2, 1})); in RunTest()
962 expected_r4.Relayout(LayoutUtil::MakeLayout({3, 0, 2, 1})); in RunTest()
1020 input_r4.Relayout(LayoutUtil::MakeLayout({0, 3, 2, 1})); in RunTest()
1032 expected_r4.Relayout(LayoutUtil::MakeLayout({0, 3, 2, 1})); in RunTest()
[all …]
|
| D | reduce_hlo_test.cc | 110 reduce_input.Relayout(reduce_input_shape->layout()); in XLA_TEST_P()
|
| D | reduce_test.cc | 182 input_literal.Relayout(LayoutUtil::MakeLayout({minor, major})); in RunR2ToR1PredTest()
215 input_literal.Relayout(LayoutUtil::MakeLayout({minor, major})); in RunR2ToR0Test()
242 input_literal.Relayout(LayoutUtil::MakeLayout({minor, major})); in RunR2ToR1Test()
299 input_literal.Relayout(LayoutUtil::MakeLayout({minor, major})); in RunVectorizedReduceTestForType()
452 input_literal = input_literal.Relayout(LayoutUtil::MakeLayout({0, 1})); in XLA_TEST_F()
483 input_literal = input_literal.Relayout(LayoutUtil::MakeLayout({0, 1})); in XLA_TEST_F()
821 input_literal.Relayout(LayoutUtil::MakeLayout(GetParam().layout)); in XLA_TEST_P()
|
| D | reshape_test.cc | 892 .Relayout(LayoutUtil::MakeLayout({3, 2, 1, 0})); in XLA_TEST_P()
920 .Relayout(LayoutUtil::MakeLayout({3, 2, 1, 0})); in XLA_TEST_P()
948 .Relayout(LayoutUtil::MakeLayout({3, 2, 1, 0})); in XLA_TEST_P()
977 .Relayout(LayoutUtil::MakeLayout({3, 2, 1, 0})); in XLA_TEST_P()
1005 .Relayout(input_literal.shape().layout()); in XLA_TEST_P()
|
| D | pad_test.cc | 211 input = input.Relayout(layout); in TEST_P()
257 input = input.Relayout(layout); in XLA_TEST_P()
|
| D | broadcast_simple_test.cc | 63 auto r3_data = LiteralUtil::CreateR3FromArray3D(*r3_array).Relayout( in MakeR3Data()
77 auto r2_data = LiteralUtil::CreateR2FromArray2D(*r2_array).Relayout( in MakeR2Data()
|
| D | cpu_gpu_fusion_test.cc | 811 Literal l1 = LiteralUtil::CreateR2FromArray2D(arr).Relayout( in XLA_TEST_F()
814 Literal l2 = LiteralUtil::CreateR2FromArray2D(arr).Relayout( in XLA_TEST_F()
|
| D | client_library_test_base.cc | 242 literal.Relayout(LayoutUtil::MakeLayout(minor_to_major)); in ComputeAndCompareLiteralWithAllInputLayouts()
|
| /external/tensorflow/tensorflow/compiler/xla/service/ |
| D | generic_transfer_manager.cc | 128 relayed_out_literal = subliteral.Relayout(device_subshape.layout(), in TransferLiteralToDeviceAsync()
|
| D | service.cc | 943 result_literal.Relayout(return_shape).ToProto(); in TransferToClient()
1089 result_literal = result_literal.Relayout(*output_layout); in ComputeConstantGraph()
|
| D | hlo_evaluator_test.cc | 2119 auto input = fft_c64x2x4x8_.Relayout(LayoutUtil::MakeLayout({0, 2, 1}));
2135 auto input = fft_c64x2x4x8_.Relayout(LayoutUtil::MakeLayout({2, 0, 1}));
2152 auto input = fft_c64x2x4x8_.Relayout(LayoutUtil::MakeLayout({1, 2, 0}));
|
| D | hlo_evaluator.cc | 2549 evaluated_.at(hlo) = evaluated_.at(hlo).Relayout(hlo->shape()); in Postprocess()
|
| D | hlo_instructions.cc | 1295 *literal_ = literal_->Relayout(new_layout, shape_index); in RelayoutConstant()
|
| D | hlo_parser.cc | 3288 *literal = literal->Relayout(shape.layout()); in ParseDenseLiteral()
|
| /external/tensorflow/tensorflow/compiler/xla/ |
| D | literal_test.cc | 649 auto data01 = data.Relayout(layout01); in TYPED_TEST()
653 auto data10 = data.Relayout(layout10); in TYPED_TEST()
772 literal_r4_2x2x3x3_dim0minor_.Relayout(layout_r4_dim0major_); in TEST_F()
776 literal_r4_2x2x3x3_dim0major_.Relayout(layout_r4_dim0minor_); in TEST_F()
788 auto relaid_mat_to_dim0major = mat_dim0minor.Relayout(layout_r2_dim0major_); in TEST_F()
799 auto relaid_mat_to_dim0minor = mat_dim0major.Relayout(layout_r2_dim0minor_); in TEST_F()
827 auto relaid_lit_to_dim0major = lit_dim0minor.Relayout(layout_r3_dim0major_); in TEST_F()
840 auto relaid_lit_to_dim0minor = lit_dim0major.Relayout(layout_r3_dim0minor_); in TEST_F()
|
| D | literal.h | 289 Literal Relayout(const Layout& new_layout,
294 Literal Relayout(const Shape& shape_with_layout) const;
|
| D | literal.cc | 708 Literal LiteralBase::Relayout(const Layout& new_layout, in Relayout() function in xla::LiteralBase
720 Literal LiteralBase::Relayout(const Shape& shape_with_layout) const { in Relayout() function in xla::LiteralBase
828 output = Relayout(LayoutUtil::GetDefaultLayoutForRank(shape().rank())); in Reshape()
|