Home
last modified time | relevance | path

Searched defs:allReduceMeshAxes (Results 1 – 2 of 2) sorted by relevance

/llvm-project/mlir/lib/Dialect/Linalg/Transforms/
H A DMeshShardingInterfaceImpl.cpp190 SmallVector<MeshAxis> allReduceMeshAxes; createAllReduceForResultWithoutPartialSharding() local
/llvm-project/mlir/lib/Dialect/Mesh/Transforms/
H A DSpmdization.cpp77 llvm::SmallVector<MeshAxis> allReduceMeshAxes; in handlePartialAxesDuringResharding() local