Searched defs:new_while (Results 1 – 9 of 9) sorted by relevance
646 HloInstruction* new_while = FindFirstWhile(m.get()); in TEST_F() local732 HloInstruction* new_while = FindFirstWhile(m.get()); in TEST_F() local801 HloInstruction* new_while = FindFirstWhile(m.get()); in TEST_F() local869 HloInstruction* new_while = FindFirstWhile(m.get()); in TEST_F() local
126 HloInstruction* new_while = containing_computation->AddInstruction( in MakeInstructionsLiveIn() local
1218 auto* new_while = computation->AddInstruction(HloInstruction::CreateWhile( in TryMergeInductionVariables() local
193 auto new_while = in HandleWhileOp() local314 auto new_while = builder.create<TF::WhileRegionOp>( in HandleWhileRegionOp() local
212 auto new_while = in HandleWhileOp() local
625 auto new_while = in HandleWhileOp() local
859 auto new_while = builder.create<TF::WhileOp>( in HandleWhileLoop() local
1733 auto new_while = builder.create<TF::WhileRegionOp>( in matchAndRewrite() local
664 auto new_while = builder.create<WhileOp>(region_while.getLoc(), in RewriteRegionWhileOp() local