Searched refs:reassociations (Results 1 – 2 of 2) sorted by relevance
| /external/llvm-project/mlir/lib/Dialect/Linalg/Transforms/ |
| D | DropUnitDims.cpp | 252 SmallVector<AffineExpr, 2> reassociations; in replaceUnitExtents() local
266 reassociations.push_back(getAffineDimExpr(dim++, context)); in replaceUnitExtents()
268 reassociations.push_back(getAffineDimExpr(dim, context)); in replaceUnitExtents()
274 reassociations.push_back(getAffineDimExpr(dim, context)); in replaceUnitExtents()
277 origRank, /*numSymbols = */ 0, reassociations, context))); in replaceUnitExtents()
278 reassociations.clear(); in replaceUnitExtents()
|
| /external/llvm-project/mlir/lib/Dialect/Linalg/IR/ |
| D | LinalgOps.cpp | 586 SmallVector<AffineExpr, 4> reassociations; in collapseReassociationMaps() local
593 reassociations.push_back(getAffineDimExpr(currDim++, context)); in collapseReassociationMaps()
597 numLhsDims, /*numSymbols =*/0, reassociations, context))); in collapseReassociationMaps()
598 reassociations.clear(); in collapseReassociationMaps()
|