Lines Matching defs:targetShardOp

44       if (auto targetShardOp = llvm::dyn_cast<ShardOp>(user)) {
45         if (targetShardOp.getAnnotateForUsers() &&
47                         targetShardOp,
49                             targetShardOp.getSharding().getDefiningOp())
62       auto targetShardOp = llvm::dyn_cast<ShardOp>(user);
63       if (!targetShardOp || !targetShardOp.getAnnotateForUsers() ||
65               targetShardOp,
66               cast<ShardingOp>(targetShardOp.getSharding().getDefiningOp())
79           reshard(builder, mesh, op, targetShardOp, sourceShard);
83                   targetShardOp.getResult().getType(), targetShard)
85       rewriter.replaceAllUsesWith(targetShardOp.getResult(),