Home
last modified time | relevance | path

Searched refs:tuple_param0 (Results 1 – 3 of 3) sorted by relevance

/external/tensorflow/tensorflow/compiler/xla/service/
Dbuffer_liveness_test.cc482 auto tuple_param0 = builder.AddInstruction(HloInstruction::CreateParameter( in TEST_F() local
492 ShapeUtil::GetSubshape(tuple_param0->shape(), {0}); in TEST_F()
495 tuple_element0_shape, tuple_param0, 0)); in TEST_F()
504 ShapeUtil::GetSubshape(tuple_param0->shape(), {1}); in TEST_F()
507 tuple_element1_shape, tuple_param0, 1)); in TEST_F()
538 TupleElementsMayInterfere(*liveness, tuple_param0, tuple_root, {0})); in TEST_F()
540 TupleElementsMayInterfere(*liveness, tuple_param0, tuple_root, {1})); in TEST_F()
546 auto tuple_param0 = builder.AddInstruction(HloInstruction::CreateParameter( in TEST_F() local
556 ShapeUtil::GetSubshape(tuple_param0->shape(), {0}); in TEST_F()
559 tuple_element0_shape, tuple_param0, 0)); in TEST_F()
[all …]
Dtuple_points_to_analysis_test.cc633 auto tuple_param0 = builder.AddInstruction( in Run() local
637 HloInstruction::CreateGetTupleElement(update_shape, tuple_param0, 1)); in Run()
645 HloInstruction::CreateGetTupleElement(input_shape, tuple_param0, 0)); in Run()
658 HloInstruction::CreateGetTupleElement(starts_shape, tuple_param0, 2)); in Run()
671 EXPECT_THAT(fusion, op::Fusion(tuple_param0)); in Run()
676 auto* fusion_param = GetFusionParameterForOperand(fusion, tuple_param0); in Run()
Dbuffer_assignment_test.cc1516 auto tuple_param0 = builder.AddInstruction( in TEST_F() local
1524 pred_param, tuple_param0, tuple_param1)); in TEST_F()
1544 assignment->GetUniqueSlice(tuple_param0, /*index=*/{0}) in TEST_F()