Searched defs:targetShard (Results 1 – 2 of 2) sorted by relevance
| /llvm-project/mlir/test/lib/Dialect/Mesh/ | ||
| H A D | TestReshardingSpmdization.cpp | 74 TypedValue<ShapedType> targetShard = matchAndRewrite() local |
| /llvm-project/mlir/lib/Dialect/Mesh/Transforms/ | ||
| H A D | Spmdization.cpp | 136 TypedValue<ShapedType> targetShard = cast<TypedValue<ShapedType>>( in splitLastAxisInResharding() local |