Searched defs:collapsingInfo (Results 1 – 1 of 1) sorted by relevance
| /llvm-project/mlir/lib/Dialect/Linalg/Transforms/ |
| H A D | ElementwiseOpFusion.cpp | 1408 getCollapsedOpIteratorTypes(ArrayRef<utils::IteratorType> iteratorTypes,const CollapsingInfo & collapsingInfo) getCollapsedOpIteratorTypes() argument
1426 getCollapsedOpIndexingMap(AffineMap indexingMap,const CollapsingInfo & collapsingInfo) getCollapsedOpIndexingMap() argument
1451 getOperandReassociation(AffineMap indexingMap,const CollapsingInfo & collapsingInfo) getOperandReassociation() argument
1479 getCollapsedOpOperand(Location loc,LinalgOp op,OpOperand * opOperand,const CollapsingInfo & collapsingInfo,OpBuilder & builder) getCollapsedOpOperand() argument
1506 generateCollapsedIndexingRegion(Location loc,Block * block,const CollapsingInfo & collapsingInfo,ValueRange loopRange,RewriterBase & rewriter) generateCollapsedIndexingRegion() argument
1544 collapseOperandsAndResults(LinalgOp op,const CollapsingInfo & collapsingInfo,RewriterBase & rewriter,SmallVectorImpl<Value> & inputOperands,SmallVectorImpl<Value> & outputOperands,SmallVectorImpl<Type> & resultTypes) collapseOperandsAndResults() argument
1573 cloneToCollapsedOp(RewriterBase & rewriter,OpTy origOp,const CollapsingInfo & collapsingInfo) cloneToCollapsedOp() argument
1581 cloneToCollapsedOp(RewriterBase & rewriter,LinalgOp origOp,const CollapsingInfo & collapsingInfo) cloneToCollapsedOp() argument
1596 cloneToCollapsedOp(RewriterBase & rewriter,GenericOp origOp,const CollapsingInfo & collapsingInfo) cloneToCollapsedOp() argument
1619 createCollapsedOp(LinalgOp op,const CollapsingInfo & collapsingInfo,RewriterBase & rewriter) createCollapsedOp() argument
1652 CollapsingInfo collapsingInfo; collapseOpIterationDims() local
[all...] |