Home
last modified time | relevance | path

Searched refs:getDimPosition (Results 1 – 8 of 8) sorted by relevance

/external/llvm-project/mlir/lib/Dialect/Linalg/Transforms/
DSparsification.cpp292 unsigned idx = map.getDimPosition(d); in findSparseAnnotations()
340 unsigned f = map.getDimPosition(d - 1); in computeIterationGraph()
341 unsigned t = map.getDimPosition(d); in computeIterationGraph()
469 unsigned i = map.getDimPosition(d); in genBuffers()
527 unsigned idx = map.getDimPosition(i); in genTensorLoad()
547 unsigned idx = map.getDimPosition(i); in genTensorStore()
605 unsigned idx = map.getDimPosition(i); in genInvariants()
DFusionOnTensors.cpp469 expandedShape.slice(foldedDims.getDimPosition(0), numFoldedDims[pos]); in fuseWithReshapeByExpansion()
/external/llvm-project/mlir/include/mlir/Dialect/SDBM/
DSDBM.h140 unsigned getDimPosition(unsigned position) const { return 1 + position; } in getDimPosition() function
/external/llvm-project/mlir/include/mlir/IR/
DAffineMap.h130 unsigned getDimPosition(unsigned idx) const;
/external/llvm-project/mlir/lib/Dialect/SDBM/
DSDBM.cpp220 return result.getDimPosition(expr.getPosition()); in get()
/external/llvm-project/mlir/lib/Dialect/Vector/
DVectorTransforms.cpp52 int64_t idx = map.getDimPosition(i); in getResultIndex()
78 int64_t idx = map.getDimPosition(i); in adjustMap()
2018 iterIndex = iMap[0].getDimPosition(lhsIndex); in lowerParallel()
2019 assert((rhsIndex < 0 || iterIndex == iMap[1].getDimPosition(rhsIndex)) && in lowerParallel()
2024 iterIndex = iMap[1].getDimPosition(rhsIndex); in lowerParallel()
DVectorOps.cpp339 unsigned pos = map.getDimPosition(idx); in verifyOutputShape()
788 auto pos = permutationMap.getDimPosition(idx); in foldExtractOpFromInsertChainAndTranspose()
/external/llvm-project/mlir/lib/IR/
DAffineMap.cpp230 unsigned AffineMap::getDimPosition(unsigned idx) const { in getDimPosition() function in AffineMap