Searched defs:meshOp (Results 1 – 2 of 2) sorted by relevance
| /llvm-project/mlir/include/mlir/Dialect/Mesh/IR/ |
| H A D | MeshOps.h | 69 mesh::MeshOp meshOp = getMeshOrNull(op, meshSymbol, symbolTableCollection); getMesh() local
|
| /llvm-project/mlir/lib/Dialect/Linalg/Transforms/ |
| H A D | MeshShardingInterfaceImpl.cpp | 134 createDestinationPassingStyleInitOperand(LinalgOp op,Value spmdizedOperand,ArrayRef<MeshAxis> reductionMeshAxes,MeshOp meshOp,ImplicitLocOpBuilder & builder) createDestinationPassingStyleInitOperand() argument
171 createDestinationPassingStyleInitOperands(LinalgOp op,MeshOp meshOp,ArrayRef<Value> spmdizedOperands,ArrayRef<MeshAxis> reductionMeshAxes,IRMapping & spmdizationMap,ImplicitLocOpBuilder & builder) createDestinationPassingStyleInitOperands() argument
|