HomeSort by relevance Sort by last modified time
    Searched refs:tuple_param0 (Results 1 - 3 of 3) sorted by null

  /external/tensorflow/tensorflow/compiler/xla/service/
buffer_liveness_test.cc 478 auto tuple_param0 = builder.AddInstruction(HloInstruction::CreateParameter( local
486 // Add(GetTupleElement(tuple_param0, 0), const0)
488 ShapeUtil::GetSubshape(tuple_param0->shape(), {0});
491 tuple_element0_shape, tuple_param0, 0));
498 // Add(GetTupleElement(tuple_param0, 1), const1)
500 ShapeUtil::GetSubshape(tuple_param0->shape(), {1});
503 tuple_element1_shape, tuple_param0, 1));
534 TupleElementsMayInterfere(*liveness, tuple_param0, tuple_root, {0}));
536 TupleElementsMayInterfere(*liveness, tuple_param0, tuple_root, {1}));
542 auto tuple_param0 = builder.AddInstruction(HloInstruction::CreateParameter local
619 auto tuple_param0 = builder.AddInstruction(HloInstruction::CreateParameter( local
749 auto tuple_param0 = builder.AddInstruction(HloInstruction::CreateParameter( local
    [all...]
tuple_points_to_analysis_test.cc 615 auto tuple_param0 = builder.AddInstruction( local
617 // Create 'tuple_element1' = GetTupleElement(tuple_param0, 1).
619 HloInstruction::CreateGetTupleElement(update_shape, tuple_param0, 1));
622 // Create 'update' = Add(GetTupleElement(tuple_param0, 1), ones)
625 // Create 'input' = GetTupleElement(tuple_param0, 0).
627 HloInstruction::CreateGetTupleElement(input_shape, tuple_param0, 0));
638 // Create slice 'starts' = GetTupleElement(tuple_param0, 2).
640 HloInstruction::CreateGetTupleElement(starts_shape, tuple_param0, 2));
653 EXPECT_THAT(fusion, op::Fusion(tuple_param0));
657 // Check points-to set of fusion parameter associated with 'tuple_param0'
    [all...]
buffer_assignment_test.cc 1347 auto tuple_param0 = builder.AddInstruction( local
    [all...]

Completed in 539 milliseconds