Searched refs:new_gte (Results 1 – 3 of 3) sorted by relevance
168 HloInstruction* new_gte = TupleUtil::AddGetTupleElements({p0, {1, 0}}); in TEST_F() local170 EXPECT_THAT(new_gte, op::GetTupleElement(existing_gte, 0)); in TEST_F()
1760 HloInstruction* new_gte = in CloneAndAppendInstructionIntoCalledComputation() local1763 TF_CHECK_OK(old_gte->ReplaceAllUsesWith(new_gte)); in CloneAndAppendInstructionIntoCalledComputation()1770 HloInstruction* new_gte = in CloneAndAppendInstructionIntoCalledComputation() local1773 TF_CHECK_OK(instruction_to_append->ReplaceAllUsesWith(new_gte)); in CloneAndAppendInstructionIntoCalledComputation()
176 TF_ASSIGN_OR_RETURN(HloInstruction * new_gte, in FuseConvertToFloat()178 TF_RETURN_IF_ERROR(comp->ReplaceInstruction(instr, new_gte)); in FuseConvertToFloat()