Searched defs:while_instr (Results 1 – 6 of 6) sorted by relevance
72 HloInstruction* while_instr = entry_computation->root_instruction(); in TEST() local103 HloInstruction* while_instr = entry_computation->root_instruction(); in TEST() local199 HloInstruction* while_instr = main->root_instruction(); in TEST() local
70 HloInstruction* while_instr = nullptr; in TEST() local131 HloInstruction* while_instr = nullptr; in TEST() local
41 HloInstruction* while_instr, HloInstruction* to_hoist) { in CreateLoopInvariantCopy()127 HloInstruction* while_instr) { in TryHoistingInvariantInstructionsFromWhileBody()
98 HloInstruction* while_instr, in MakeInstructionsLiveIn()244 HloInstruction* while_instr = in MakeCountedLoop() local
50 HloInstruction* while_instr) { in TrySinkingConstantsIntoWhileLoop()
122 auto while_instr = *maybe_while; in GetAllTuples() local