Searched defs:new_instruction (Results 1 – 9 of 9) sorted by relevance
63 HloInstruction* new_instruction) { in replace_instruction()159 HloInstruction* new_instruction) { in replace_instruction()
45 HloInstruction* new_instruction) { in MapInstruction()
172 auto new_instruction = in ExpandInstruction() local
282 std::unique_ptr<HloInstruction> new_instruction) { in ReplaceWithNewInstruction()296 HloInstruction* new_instruction) { in ReplaceInstruction()
74 HloInstruction* new_instruction = in CreateLoopInvariantCopy() local
868 std::unique_ptr<HloInstruction> new_instruction) { in ReplaceWithNewInstruction()875 std::unique_ptr<HloInstruction> new_instruction) { in ReplaceWithNewEntryComputationParameter()881 HloInstruction* new_instruction) { in ReplaceInstruction()
1245 HloInstruction* new_instruction, HloInstruction* tuple, in ReplaceTupleWith()1495 HloInstruction* new_instruction, HloInstructionSequence* new_sequence, in EnsureInstructionAndOperandsInserted()
567 HloInstruction* old_instruction, HloInstruction* new_instruction) { in ReplaceInstructionIfSameShape()
88 if (auto new_instruction = in FinishVisit() local