Searched defs:new_while_op (Results 1 – 8 of 8) sorted by relevance
45 HloInstruction* new_while_op) { in CopyFrontendAttributes()171 auto* new_while_op = computation->AddInstruction(HloInstruction::CreateWhile( in RemoveDeadTupleIndices() local754 auto* new_while_op = computation->AddInstruction(HloInstruction::CreateWhile( in TryRemoveConstantParams() local1085 auto* new_while_op = computation->AddInstruction(HloInstruction::CreateWhile( in TryFlattenNestedTuples() local
461 HloInstruction* new_while_op = in TEST_F() local516 HloInstruction* new_while_op = in TEST_F() local
294 auto new_while_op = builder.create<T>(while_op->getLoc(), new_operand_types, in GetUpdatedWhileOp() local390 auto new_while_op = GetUpdatedWhileOp( in FreezeVariables() local
360 auto new_while_op = builder.create<TF::WhileOp>( in RewriteWhileOp() local
113 auto new_while_op = builder.create<mhlo::WhileOp>(while_op.getLoc(), in prepareWhileOp() local
275 auto new_while_op = OpBuilder(while_op).create<WhileOp>( in ReduceWhileOperands() local
270 auto new_while_op = OpBuilder(while_op).create<WhileOp>( in OutlineWhile() local
3243 auto new_while_op = rewriter.create<WhileRegionOp>( in matchAndRewrite() local