Lines Matching defs:newShardOp

287   auto newShardOp =
292       operandValue, newShardOp, [operandOp, operandValue](OpOperand &use) {
301       builder.create<ShardOp>(operandValue.getLoc(), newShardOp, shardingOp,
303   rewriter.replaceAllUsesExcept(newShardOp, newShardOp2, newShardOp2);
333   auto newShardOp =
338       operandValue, newShardOp, [operandOp, operandValue](OpOperand &use) {
347   builder.setInsertionPoint(newShardOp);
352       newShardOp.getSrc(), newPreceedingShardOp, [&newShardOp](OpOperand &use) {
353         return use.getOwner() == newShardOp.getOperation();