Searched defs:new_while_op (Results 1 – 5 of 5) sorted by relevance
43 HloInstruction* new_while_op) { in CopyBackendConfig()51 HloInstruction* new_while_op) { in CopyFrontendAttributes()175 auto* new_while_op = computation->AddInstruction(HloInstruction::CreateWhile( in RemoveDeadTupleIndices() local649 auto* new_while_op = computation->AddInstruction(HloInstruction::CreateWhile( in TryRemoveConstantParams() local979 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
307 auto new_while_op = builder.create<T>(while_op->getLoc(), new_operand_types, in GetUpdatedWhileOp() local381 auto new_while_op = in runOnOperation() local386 auto new_while_op = GetUpdatedWhileOp(while_op, items.getSecond()); in runOnOperation() local
279 auto new_while_op = OpBuilder(while_op).create<WhileOp>( in OutlineWhile() local
3106 auto new_while_op = rewriter.create<WhileRegionOp>( in matchAndRewrite() local