Home
last modified time | relevance | path

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

/llvm-project-15.0.7/mlir/lib/Dialect/Vector/Transforms/
H A DVectorDropLeadUnitDim.cpp310 int64_t dimToDrop = oldIndexingMaps[2].getDimPosition(0); in matchAndRewrite()
333 int64_t orginalZeroDim = it.value().getDimPosition(0); in matchAndRewrite()
345 int64_t currDim = map.getDimPosition(i); in matchAndRewrite()
370 if (map.getDimPosition(0) == dimToDrop) in matchAndRewrite()
374 int64_t currDim = map.getDimPosition(i); in matchAndRewrite()
H A DVectorTransforms.cpp47 int64_t idx = map.getDimPosition(i); in getResultIndex()
73 int64_t idx = map.getDimPosition(i); in adjustMap()
184 if (isReductionIterator(iteratorTypes[map.getDimPosition(i)])) in getReductionIndex()
192 static llvm::Optional<unsigned> getDimPosition(AffineMap map, unsigned dim) { in getDimPosition() function
194 if (map.getDimPosition(i) == dim) in getDimPosition()
648 getDimPosition(lhsMap, accMap.getDimPosition(i)); in matchAndRewrite()
658 getDimPosition(rhsMap, accMap.getDimPosition(i)); in matchAndRewrite()
1133 .getValue()[map.getDimPosition(i)])) { in matchAndRewrite()
1875 iterIndex = iMap[0].getDimPosition(lhsIndex); in lowerParallel()
1876 if (rhsIndex >= 0 && iterIndex != iMap[1].getDimPosition(rhsIndex)) in lowerParallel()
[all …]
H A DVectorUnrollDistribute.cpp588 map[affineMap.getDimPosition(i)] = extract.getResultType().getDimSize(i); in matchAndRewrite()
597 unsigned dim = it.value().getDimPosition(i); in matchAndRewrite()
H A DVectorDistribute.cpp347 unsigned position = map.getDimPosition(i); in tryDistributeOp()
/llvm-project-15.0.7/mlir/lib/IR/
H A DAffineMap.cpp315 unsigned AffineMap::getDimPosition(unsigned idx) const { in getDimPosition() function in AffineMap
322 if (getDimPosition(i) == input) in getPermutedPosition()
693 exprs[map.getDimPosition(i)] = getAffineDimExpr(i, context); in inverseAndBroadcastProjectedPermutation()
/llvm-project-15.0.7/mlir/lib/Dialect/Linalg/Transforms/
H A DSplitReduction.cpp126 unsigned dim = map.getDimPosition(idx); in splitReduction()
171 unsigned dim = oldOutputMap.getDimPosition(oldDim); in splitReduction()
H A DFusionOnTensors.cpp93 return tiledProducerIndexingSubMap.getDimPosition(idx); in getTiledProducerLoops()
H A DElementwiseOpFusion.cpp535 expandedShape.slice(foldedDims.getDimPosition(0), numExpandedDims[pos]); in compute()
/llvm-project-15.0.7/mlir/include/mlir/IR/
H A DAffineMap.h167 unsigned getDimPosition(unsigned idx) const;
/llvm-project-15.0.7/mlir/lib/Dialect/SparseTensor/Transforms/
H A DSparsification.cpp124 inv[i] = m.getDimPosition(i); in permute()
138 return order.getDimPosition(d); in perm()
H A DSparseTensorConversion.cpp218 rev[p.getDimPosition(i)] = constantIndex(builder, loc, i); in newParams()
/llvm-project-15.0.7/mlir/lib/Dialect/Vector/IR/
H A DVectorOps.cpp691 unsigned pos = map.getDimPosition(idx); in verifyOutputShape()