Lines Matching defs:shardingOp
218 ShardingInterface shardingOp,
228 shardingOp.getShardingOption(operandShardings, resultShardings);
238 shardingOp.getShardingAnnotations(*shardingOption);
247 getReshardingRquirementKind(shardingOp, *operandAndResultShardings);
289 ShardingInterface shardingOp = llvm::dyn_cast<ShardingInterface>(op);
290 if (!shardingOp) {
339 shardingOp, possibleOperandShardingAttrs, possibleResultShardingAttrs);
352 if (failed(shardingOp.addShardingAnnotations(builder, *shardingOption))) {
380 if (auto shardingOp = llvm::dyn_cast<ShardingInterface>(&op))
381 shardingOp.printLoopTypesAndIndexingMaps(llvm::dbgs());