Searched refs:hoisted_instructions (Results 1 – 1 of 1) sorted by relevance
39 flat_hash_map<HloInstruction*, HloInstruction*>* hoisted_instructions, in CreateLoopInvariantCopy() argument67 : FindOrDie(*hoisted_instructions, old_operand); in CreateLoopInvariantCopy()78 InsertOrDie(hoisted_instructions, old_instruction, new_instruction); in CreateLoopInvariantCopy()92 if (hoisted_instructions->contains(next_operand) || in CreateLoopInvariantCopy()158 flat_hash_map<HloInstruction*, HloInstruction*> hoisted_instructions; in TryHoistingInvariantInstructionsFromWhileBody() local243 return hoisted_instructions.find(op) != hoisted_instructions.end() || in TryHoistingInvariantInstructionsFromWhileBody()266 CreateLoopInvariantCopy(&hoisted_instructions, in TryHoistingInvariantInstructionsFromWhileBody()272 FindOrDie(hoisted_instructions, instruction)); in TryHoistingInvariantInstructionsFromWhileBody()