Searched defs:new_while (Results 1 – 8 of 8) sorted by relevance
609 HloInstruction* new_while = FindFirstWhile(m.get()); in TEST_F() local695 HloInstruction* new_while = FindFirstWhile(m.get()); in TEST_F() local762 HloInstruction* new_while = FindFirstWhile(m.get()); in TEST_F() local830 HloInstruction* new_while = FindFirstWhile(m.get()); in TEST_F() local
121 HloInstruction* new_while = containing_computation->AddInstruction( in MakeInstructionsLiveIn() local
1177 auto* new_while = computation->AddInstruction(HloInstruction::CreateWhile( in TryMergeInductionVariables() local
209 auto new_while = in HandleWhileOp() local330 auto new_while = builder.create<TF::WhileRegionOp>( in HandleWhileRegionOp() local
205 auto new_while = in HandleWhileOp() local
626 auto new_while = in HandleWhileOp() local
930 auto new_while = builder.create<TF::WhileOp>( in HandleWhileLoop() local
656 auto new_while = builder.create<WhileOp>(region_while.getLoc(), in RewriteRegionWhileOp() local