| /external/tensorflow/tensorflow/compiler/xla/tests/ |
| D | reduce_test.cc | 181 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()
|
| D | convolution_test.cc | 780 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 …]
|
| D | reshape_test.cc | 850 .Relayout(LayoutUtil::MakeLayout({3, 2, 1, 0})); in XLA_TEST_P()
877 .Relayout(LayoutUtil::MakeLayout({3, 2, 1, 0})); in XLA_TEST_P()
904 .Relayout(LayoutUtil::MakeLayout({3, 2, 1, 0})); in XLA_TEST_P()
932 .Relayout(LayoutUtil::MakeLayout({3, 2, 1, 0})); in XLA_TEST_P()
959 .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 | 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/ |
| D | literal_test.cc | 653 auto data01 = data.Relayout(layout01); in TYPED_TEST()
657 auto data10 = data.Relayout(layout10); in TYPED_TEST()
735 literal_r4_2x2x3x3_dim0minor_.Relayout(layout_r4_dim0major_); in TEST_F()
739 literal_r4_2x2x3x3_dim0major_.Relayout(layout_r4_dim0minor_); in TEST_F()
751 auto relaid_mat_to_dim0major = mat_dim0minor.Relayout(layout_r2_dim0major_); in TEST_F()
762 auto relaid_mat_to_dim0minor = mat_dim0major.Relayout(layout_r2_dim0minor_); in TEST_F()
790 auto relaid_lit_to_dim0major = lit_dim0minor.Relayout(layout_r3_dim0major_); in TEST_F()
803 auto relaid_lit_to_dim0minor = lit_dim0major.Relayout(layout_r3_dim0minor_); in TEST_F()
|
| D | literal.h | 266 Literal Relayout(const Layout& new_layout,
271 Literal Relayout(const Shape& shape_with_layout) const;
|
| D | literal.cc | 578 Literal LiteralBase::Relayout(const Layout& new_layout, in Relayout() function in xla::LiteralBase
590 Literal LiteralBase::Relayout(const Shape& shape_with_layout) const { in Relayout() function in xla::LiteralBase
655 output = Relayout(LayoutUtil::GetDefaultLayoutForRank(shape().rank())); in Reshape()
|
| /external/tensorflow/tensorflow/compiler/xla/service/ |
| D | generic_transfer_manager.cc | 140 relayed_out_literal = subliteral.Relayout(device_subshape.layout(), in TransferLiteralToDeviceAsync()
|
| D | service.cc | 966 result_literal.Relayout(return_shape).ToProto(); in TransferToClient()
1112 result_literal = result_literal.Relayout(*output_layout); in ComputeConstantGraph()
|
| D | hlo_evaluator.cc | 1609 evaluated_.at(hlo) = evaluated_.at(hlo).Relayout(hlo->shape()); in Postprocess()
|
| D | hlo_instructions.cc | 1055 *literal_ = literal_->Relayout(new_layout, shape_index); in RelayoutConstant()
|
| D | hlo_parser.cc | 2327 *literal = literal->Relayout(shape.layout()); in ParseDenseLiteral()
|
| /external/tensorflow/tensorflow/compiler/xla/python/ |
| D | local_computation_builder.cc | 110 Literal relaid = argument.Relayout(shape_with_layout.value()); in FromLiteral()
|