Home
last modified time | relevance | path

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

/external/tensorflow/tensorflow/compiler/xla/service/
Dwhile_loop_invariant_code_motion.cc40 flat_hash_set<HloInstruction*>* unhoisted_invariant_instructions, in CreateLoopInvariantCopy() argument
83 CHECK_EQ(unhoisted_invariant_instructions->erase(old_instruction), in CreateLoopInvariantCopy()
164 flat_hash_set<HloInstruction*> unhoisted_invariant_instructions; in TryHoistingInvariantInstructionsFromWhileBody() local
174 InsertOrDie(&unhoisted_invariant_instructions, instr); in TryHoistingInvariantInstructionsFromWhileBody()
178 if (unhoisted_invariant_instructions.empty() && !hoist_constants_) { in TryHoistingInvariantInstructionsFromWhileBody()
244 unhoisted_invariant_instructions.contains(op) || in TryHoistingInvariantInstructionsFromWhileBody()
259 InsertOrDie(&unhoisted_invariant_instructions, instruction); in TryHoistingInvariantInstructionsFromWhileBody()
267 &unhoisted_invariant_instructions, while_instr, in TryHoistingInvariantInstructionsFromWhileBody()