Searched refs:originalShape (Results 1 – 4 of 4) sorted by relevance
| /llvm-project-15.0.7/mlir/lib/Dialect/Vector/Transforms/ |
| H A D | VectorTransferPermutationMapRewritePatterns.cpp | 80 ArrayRef<int64_t> originalShape = op.getVectorType().getShape(); in matchAndRewrite() local
81 SmallVector<int64_t> newVectorShape(originalShape.size()); in matchAndRewrite()
83 newVectorShape[pos.value()] = originalShape[pos.index()]; in matchAndRewrite()
|
| H A D | VectorUnrollDistribute.cpp | 30 static SmallVector<int64_t, 4> getVectorOffset(ArrayRef<int64_t> originalShape, in getVectorOffset() argument
34 computeStrides(originalShape, targetShape); in getVectorOffset()
53 DecomposeShapeIterator(ArrayRef<int64_t> originalShape, in DecomposeShapeIterator() argument
58 sliceStrides(originalShape.size()) { in DecomposeShapeIterator()
59 assert(originalShape.size() == targetShape.size()); in DecomposeShapeIterator()
63 SmallVector<int64_t> sliceDimCounts(originalShape.size()); in DecomposeShapeIterator()
64 for (unsigned r = 0; r < originalShape.size(); ++r) { in DecomposeShapeIterator()
65 sliceDimCounts[r] = ceilDiv(originalShape[r], targetShape[r]); in DecomposeShapeIterator()
|
| /llvm-project-15.0.7/mlir/lib/IR/ |
| H A D | BuiltinTypes.cpp | 438 mlir::computeRankReductionMask(ArrayRef<int64_t> originalShape, in computeRankReductionMask() argument
440 size_t originalRank = originalShape.size(), reducedRank = reducedShape.size(); in computeRankReductionMask()
446 originalShape[originalIdx] == reducedShape[reducedIdx]) { in computeRankReductionMask()
454 if (originalShape[originalIdx] != 1) in computeRankReductionMask()
474 ArrayRef<int64_t> originalShape = originalShapedType.getShape(); in isRankReducedType() local
477 unsigned originalRank = originalShape.size(), in isRankReducedType()
483 computeRankReductionMask(originalShape, candidateReducedShape); in isRankReducedType()
|
| /llvm-project-15.0.7/mlir/include/mlir/IR/ |
| H A D | BuiltinTypes.h | 343 computeRankReductionMask(ArrayRef<int64_t> originalShape,
|