Searched refs:inner_tuple (Results 1 – 8 of 8) sorted by relevance
| /external/tensorflow/tensorflow/compiler/xla/service/ |
| D | tuple_points_to_analysis_test.cc | 182 auto inner_tuple = builder.AddInstruction( in TEST_F() local
188 HloInstruction::CreateTuple({inner_tuple, constant3})); in TEST_F()
198 EXPECT_EQ(3, points_to_analysis_->GetPointsToSet(inner_tuple).size()); in TEST_F()
199 EXPECT_FALSE(points_to_analysis_->GetPointsToSet(inner_tuple).IsAmbiguous()); in TEST_F()
200 EXPECT_TRUE(points_to_analysis_->GetPointsToSet(inner_tuple).IsDistinct()); in TEST_F()
202 points_to_analysis_->GetPointsToSet(inner_tuple).CreateFlattenedSet(), in TEST_F()
203 {constant1, constant2, inner_tuple}); in TEST_F()
205 points_to_analysis_->GetPointsToSet(inner_tuple).element({}), in TEST_F()
206 {inner_tuple}); in TEST_F()
208 points_to_analysis_->GetPointsToSet(inner_tuple).tuple_sources({}), in TEST_F()
[all …]
|
| D | buffer_liveness_test.cc | 369 auto inner_tuple = in TEST_F() local
374 builder.AddInstruction(HloInstruction::CreateTuple({inner_tuple, exp})); in TEST_F()
387 EXPECT_TRUE(InstructionMaybeLiveOut(*liveness, inner_tuple)); in TEST_F()
|
| D | layout_assignment_test.cc | 299 auto inner_tuple = in TEST_F() local
302 HloInstruction::CreateTuple({inner_tuple, inner_tuple})); in TEST_F()
|
| D | copy_insertion_test.cc | 626 auto inner_tuple = in BuildNestedBodyComputation() local
628 builder.AddInstruction(HloInstruction::CreateTuple({add0, inner_tuple})); in BuildNestedBodyComputation()
|
| /external/tensorflow/tensorflow/compiler/xla/tests/ |
| D | deallocation_test.cc | 110 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()
|
| D | local_client_execute_test.cc | 224 auto inner_tuple = Tuple(&builder, {x, y, x}); in XLA_TEST_F() local
225 Tuple(&builder, {inner_tuple, x}); in XLA_TEST_F()
337 auto inner_tuple = GetTupleElement(param, 0); in XLA_TEST_F() local
338 auto inner_array = GetTupleElement(inner_tuple, 0); in XLA_TEST_F()
339 auto inner_vector = GetTupleElement(inner_tuple, 1); in XLA_TEST_F()
|
| D | tuple_test.cc | 416 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/ |
| D | literal_test.cc | 1082 Literal inner_tuple = LiteralUtil::MakeTuple( in TEST_F() local
1084 Literal nested_tuple = LiteralUtil::MakeTuple({&matrix, &inner_tuple}); in TEST_F()
|