Lines Matching defs:opOperand
153 mesh::getMeshSharding(OpOperand &opOperand) {
154 Value val = opOperand.get();
447 static FailureOr<MeshSharding> getSharding(OpOperand &opOperand,
450 Value operandValue = opOperand.get();
479 fromArrayOfVector(opOperand.get().getContext(), splitAxes));
495 for (OpOperand &opOperand : op->getOpOperands()) {
497 opOperand, shardingOption, maps[opOperand.getOperandNumber()]);
532 static LogicalResult addShardOp(OpBuilder &b, OpOperand &opOperand,
537 getSharding(opOperand, shardingOption, map);
542 maybeInsertSourceShardingAnnotation(sharding.value(), opOperand, b);
568 for (OpOperand &opOperand : op->getOpOperands()) {
569 if (failed(addShardOp(b, opOperand, shardingOption,
570 maps[opOperand.getOperandNumber()])))