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
125 HloInstruction* new_while = containing_computation->AddInstruction( in MakeInstructionsLiveIn() local
1324 auto* new_while = computation->AddInstruction(HloInstruction::CreateWhile( in TryMergeInductionVariables() local
195 auto new_while = builder.create<TF::WhileOp>( in HandleWhileOp() local317 auto new_while = builder.create<TF::WhileRegionOp>( in HandleWhileRegionOp() local
178 auto new_while = builder.create<TF::WhileOp>( in HandleWhileOp() local
863 auto new_while = builder.create<TF::WhileOp>( in HandleWhileLoop() local
626 auto new_while = builder.create<TF::WhileOp>( in HandleWhileOp() local
3092 auto new_while = rewriter.create<TF::WhileRegionOp>( in matchAndRewrite() local
723 auto new_while = builder.create<WhileOp>(region_while.getLoc(), in RewriteRegionWhileOp() local