Home
last modified time | relevance | path

Searched refs:inner_tuple (Results 1 – 7 of 7) sorted by relevance

/external/tensorflow/tensorflow/compiler/xla/service/
Dtuple_points_to_analysis_test.cc184 auto inner_tuple = builder.AddInstruction( in TEST_F() local
190 HloInstruction::CreateTuple({inner_tuple, constant3})); in TEST_F()
200 EXPECT_EQ(3, points_to_analysis_->GetPointsToSet(inner_tuple).size()); in TEST_F()
201 EXPECT_FALSE(points_to_analysis_->GetPointsToSet(inner_tuple).IsAmbiguous()); in TEST_F()
202 EXPECT_TRUE(points_to_analysis_->GetPointsToSet(inner_tuple).IsDistinct()); in TEST_F()
204 points_to_analysis_->GetPointsToSet(inner_tuple).CreateFlattenedSet(), in TEST_F()
205 {constant1, constant2, inner_tuple}); in TEST_F()
207 points_to_analysis_->GetPointsToSet(inner_tuple).element({}), in TEST_F()
208 {inner_tuple}); in TEST_F()
210 points_to_analysis_->GetPointsToSet(inner_tuple).tuple_sources({}), in TEST_F()
[all …]
Dlayout_assignment_test.cc299 auto inner_tuple = in TEST_F() local
302 HloInstruction::CreateTuple({inner_tuple, inner_tuple})); in TEST_F()
Dcopy_insertion_test.cc632 auto inner_tuple = in BuildNestedBodyComputation() local
634 builder.AddInstruction(HloInstruction::CreateTuple({add0, inner_tuple})); in BuildNestedBodyComputation()
/external/tensorflow/tensorflow/compiler/xla/tests/
Ddeallocation_test.cc110 auto inner_tuple = in XLA_TEST_F() local
112 Tuple(&builder, {element, inner_tuple, element}); in XLA_TEST_F()
125 auto inner_tuple = in XLA_TEST_F() local
128 Tuple(&builder, {inner_tuple, ConstantR1<float>(&builder, {0.123, 0.456})}); in XLA_TEST_F()
Dlocal_client_execute_test.cc227 auto inner_tuple = Tuple(&builder, {x, y, x}); in XLA_TEST_F() local
228 Tuple(&builder, {inner_tuple, x}); in XLA_TEST_F()
340 auto inner_tuple = GetTupleElement(param, 0); in XLA_TEST_F() local
341 auto inner_array = GetTupleElement(inner_tuple, 0); in XLA_TEST_F()
342 auto inner_vector = GetTupleElement(inner_tuple, 1); in XLA_TEST_F()
Dtuple_test.cc416 auto inner_tuple = Tuple(&builder, {ConstantR1<float>(&builder, {1.0, 2.0}), in XLA_TEST_F() local
418 Tuple(&builder, {inner_tuple, ConstantR1<float>(&builder, {22.0, 44.0})}); in XLA_TEST_F()
/external/tensorflow/tensorflow/compiler/xla/
Dliteral_test.cc1151 Literal inner_tuple = LiteralUtil::MakeTuple( in TEST_F() local
1153 Literal nested_tuple = LiteralUtil::MakeTuple({&matrix, &inner_tuple}); in TEST_F()