Searched defs:shardingOp (Results 1 – 2 of 2) sorted by relevance
| /llvm-project/mlir/lib/Dialect/Mesh/Transforms/ | ||
| H A D | ShardingPropagation.cpp | 220 selectShardingOption(ShardingInterface shardingOp,ArrayRef<SmallVector<MeshShardingAttr>> possibleOperandShardingAttrs,ArrayRef<SmallVector<MeshShardingAttr>> possibleResultShardingAttrs) selectShardingOption() argument |
| /llvm-project/mlir/lib/Dialect/Mesh/Interfaces/ | ||
| H A D | ShardingInterface.cpp | 256 ShardingInterface shardingOp = llvm::cast<ShardingInterface>(op); defaultGetShardingOption() local |