Searched defs:splitCount (Results 1 – 1 of 1) sorted by relevance
| /llvm-project/mlir/lib/Dialect/Mesh/Transforms/ | ||
| H A D | Spmdization.cpp | 251 allGatherResultShapeInUnsplitLastAxis(ShapedType sourceShape,int64_t splitCount,int64_t splitTensorAxis) allGatherResultShapeInUnsplitLastAxis() argument |