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
44 HloInstruction* while_instr, HloInstruction* to_hoist) { in CreateLoopInvariantCopy()132 HloInstruction* while_instr, BoundNonLinearCompilerAnalysis* allowance) { 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
68 HloInstruction* while_instr) { in TrySinkingConstantsIntoWhileLoop()
237 auto while_instr = *maybe_while; in GetAllTuples() local
1879 auto while_instr = FindInstruction(module, "while"); in TEST_P() local1999 auto while_instr = FindInstruction(module.get(), "while"); in TEST_P() local
4235 const HloInstruction* while_instr = FindInstruction(module.get(), "while"); in TEST_P() local4289 const HloInstruction* while_instr = FindInstruction(module.get(), "while"); in TEST_P() local4429 const HloInstruction* while_instr = FindInstruction(module.get(), "while"); in TEST_P() local