Searched defs:while_instr (Results 1 – 8 of 8) sorted by relevance
77 HloInstruction* while_instr = entry_computation->root_instruction(); in TEST_F() local108 HloInstruction* while_instr = entry_computation->root_instruction(); in TEST_F() local204 HloInstruction* while_instr = main->root_instruction(); in TEST_F() local
79 HloInstruction* while_instr = nullptr; in TEST_F() local142 HloInstruction* while_instr = nullptr; in TEST_F() local
41 HloInstruction* while_instr, HloInstruction* to_hoist) { in CreateLoopInvariantCopy()127 HloInstruction* while_instr) { in TryHoistingInvariantInstructionsFromWhileBody()
55 HloInstruction* while_instr, HloInstruction* to_hoist) { in CreateLoopInvariantCopy()116 TryHoistingInvariantInstructionsFromWhileBody(HloInstruction* while_instr) { in TryHoistingInvariantInstructionsFromWhileBody()
98 HloInstruction* while_instr, in MakeInstructionsLiveIn()259 HloInstruction* while_instr = in MakeCountedLoop() local
50 HloInstruction* while_instr) { in TrySinkingConstantsIntoWhileLoop()
238 auto while_instr = *maybe_while; in GetAllTuples() local
1670 auto while_instr = FindInstruction(module, "while"); in TEST_P() local1791 auto while_instr = FindInstruction(module.get(), "while"); in TEST_P() local