Searched defs:reductionMeshAxes (Results 1 – 1 of 1) sorted by relevance
133 ArrayRef<MeshAxis> reductionMeshAxes, MeshOp meshOp, in createDestinationPassingStyleInitOperand() argument 172 createDestinationPassingStyleInitOperands(LinalgOp op,MeshOp meshOp,ArrayRef<Value> spmdizedOperands,ArrayRef<MeshAxis> reductionMeshAxes,IRMapping & spmdizationMap,ImplicitLocOpBuilder & builder) createDestinationPassingStyleInitOperands() argument 229 SmallVector<MeshAxis> reductionMeshAxes = mesh::getReductionMeshAxes( spmdizeLinalgOpWithShardedReduction() local