Lines Matching defs:shardOp

112     auto shardOp = llvm::dyn_cast<mesh::ShardOp>(user);
113     if (!shardOp)
115     return !shardOp.getAnnotateForUsers();
123     auto shardOp = llvm::cast<mesh::ShardOp>(*val.getUsers().begin());
124     return std::make_pair(false, MeshSharding(shardOp.getSharding()));
128     auto shardOp = llvm::dyn_cast<mesh::ShardOp>(user);
129     if (!shardOp)
131     return shardOp.getAnnotateForUsers();
136       ShardOp shardOp = llvm::dyn_cast<ShardOp>(user);
137       if (shardOp)
138         shardOps.push_back(shardOp);
155   if (ShardOp shardOp = val.getDefiningOp<ShardOp>())
156     return std::make_pair(shardOp.getAnnotateForUsers(),
157                           MeshSharding(shardOp.getSharding()));