Searched defs:while_instr (Results 1 – 9 of 9) 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
43 HloInstruction* while_instr, HloInstruction* to_hoist) { in CreateLoopInvariantCopy()131 HloInstruction* while_instr) { in TryHoistingInvariantInstructionsFromWhileBody()
56 HloInstruction* while_instr, HloInstruction* to_hoist) { in CreateLoopInvariantCopy()117 TryHoistingInvariantInstructionsFromWhileBody(HloInstruction* while_instr) { in TryHoistingInvariantInstructionsFromWhileBody()
102 HloInstruction* while_instr, in MakeInstructionsLiveIn()270 HloInstruction* while_instr = owned_while.get(); in MakeCountedLoop() local
50 HloInstruction* while_instr) { in TrySinkingConstantsIntoWhileLoop()
238 auto while_instr = *maybe_while; in GetAllTuples() local
1764 auto while_instr = FindInstruction(module, "while"); in TEST_P() local1885 auto while_instr = FindInstruction(module.get(), "while"); in TEST_P() local
4403 const HloInstruction* while_instr = FindInstruction(module.get(), "while"); in TEST_P() local4457 const HloInstruction* while_instr = FindInstruction(module.get(), "while"); in TEST_P() local4597 const HloInstruction* while_instr = FindInstruction(module.get(), "while"); in TEST_P() local