Home
last modified time | relevance | path

Searched refs:newShape (Results 1 – 16 of 16) sorted by relevance

/llvm-project-15.0.7/mlir/include/mlir/IR/
H A DBuiltinTypes.h176 Builder &setShape(ArrayRef<int64_t> newShape) { in setShape() argument
177 shape = newShape; in setShape()
227 Builder &setShape(ArrayRef<int64_t> newShape) { in setShape() argument
228 shape = newShape; in setShape()
293 Builder &setShape(ArrayRef<int64_t> newShape,
296 shape = newShape;
/llvm-project-15.0.7/mlir/lib/Dialect/Affine/Transforms/
H A DPipelineDataTransfer.cpp74 SmallVector<int64_t, 4> newShape(1 + oldMemRefType.getRank()); in doubleBuffer() local
75 newShape[0] = 2; in doubleBuffer()
76 std::copy(oldShape.begin(), oldShape.end(), newShape.begin() + 1); in doubleBuffer()
77 return MemRefType::Builder(oldMemRefType).setShape(newShape).setLayout({}); in doubleBuffer()
H A DLoopFusion.cpp910 SmallVector<int64_t, 4> newShape; in createPrivateMemRef() local
917 region.getConstantBoundingSizeAndShape(&newShape, &lbs, &lbDivisors); in createPrivateMemRef()
953 auto newMemRefType = MemRefType::get(newShape, oldMemRefType.getElementType(), in createPrivateMemRef()
/llvm-project-15.0.7/mlir/lib/Dialect/MemRef/Transforms/
H A DMultiBuffer.cpp112 SmallVector<int64_t, 4> newShape(1, multiplier); in multiBuffer() local
114 newShape.append(oldShape.begin(), oldShape.end()); in multiBuffer()
115 auto newMemref = MemRefType::get(newShape, allocOp.getType().getElementType(), in multiBuffer()
/llvm-project-15.0.7/mlir/lib/Dialect/Linalg/Transforms/
H A DSplitReduction.cpp121 SmallVector<int64_t> newShape; in splitReduction() local
128 newShape.push_back(ratio); in splitReduction()
129 newShape.push_back(op.getShape(operand)[idx] / ratio); in splitReduction()
136 newShape.push_back(op.getShape(operand)[idx]); in splitReduction()
144 if (newShape == op.getShape(operand)) { in splitReduction()
149 newShape, in splitReduction()
H A DDropUnitDims.cpp248 SmallVector<int64_t> newShape; in replaceUnitExtents() local
271 newShape.push_back(shape[dim]); in replaceUnitExtents()
289 replacementType = RankedTensorType::get(newShape, elementType); in replaceUnitExtents()
291 replacementType = MemRefType::get(newShape, elementType); in replaceUnitExtents()
/llvm-project-15.0.7/mlir/lib/Dialect/Vector/Transforms/
H A DVectorTransferPermutationMapRewritePatterns.cpp251 SmallVector<int64_t> newShape = llvm::to_vector<4>( in matchAndRewrite() local
254 if (newShape.empty()) in matchAndRewrite()
257 VectorType::get(newShape, originalVecType.getElementType()); in matchAndRewrite()
H A DVectorDropLeadUnitDim.cpp26 ArrayRef<int64_t> newShape = in trimLeadingOneDims() local
29 if (newShape.empty()) in trimLeadingOneDims()
30 newShape = oldShape.take_back(); in trimLeadingOneDims()
31 return VectorType::get(newShape, oldType.getElementType()); in trimLeadingOneDims()
/llvm-project-15.0.7/mlir/lib/Dialect/Bufferization/IR/
H A DBufferizationOps.cpp304 SmallVector<int64_t> newShape = llvm::to_vector(op.getType().getShape()); in matchAndRewrite() local
313 newShape[i] = intVal.getSExtValue(); in matchAndRewrite()
319 newShape, op.getType().getElementType(), op.getType().getEncoding()); in matchAndRewrite()
/llvm-project-15.0.7/mlir/lib/Dialect/Tosa/IR/
H A DTosaOps.cpp611 ArrayAttr newShape = adaptor.getNewShape(); in inferReturnTypeComponents() local
613 getI64Values(newShape, newShapeValue); in inferReturnTypeComponents()
756 llvm::SmallVector<int64_t> newShape; in inferReturnTypeComponents() local
757 getI64Values(adaptor.getOutputSize(), newShape); in inferReturnTypeComponents()
758 outputShape[1] = newShape[0]; in inferReturnTypeComponents()
759 outputShape[2] = newShape[1]; in inferReturnTypeComponents()
H A DTosaCanonicalizations.cpp87 ArrayAttr newShape = op.getNewShape(); in matchAndRewrite() local
100 llvm::map_range(newShape.getValue(), [](const Attribute &val) { in matchAndRewrite()
/llvm-project-15.0.7/mlir/lib/Dialect/Tensor/IR/
H A DTensorOps.cpp638 SmallVector<int64_t, 4> newShape; in matchAndRewrite() local
643 newShape.push_back(dim); in matchAndRewrite()
648 newShape.push_back(ShapedType::kDynamicSize); in matchAndRewrite()
652 newShape.push_back(index.getSExtValue()); in matchAndRewrite()
661 loc, RankedTensorType::get(newShape, resultType.getElementType()), in matchAndRewrite()
815 SmallVector<int64_t, 4> newShape; in computeTensorReshapeCollapsedType() local
816 newShape.reserve(reassociation.size()); in computeTensorReshapeCollapsedType()
831 newShape.push_back(size); in computeTensorReshapeCollapsedType()
835 return RankedTensorType::get(newShape, type.getElementType()); in computeTensorReshapeCollapsedType()
/llvm-project-15.0.7/mlir/lib/Dialect/Affine/Utils/
H A DUtils.cpp1782 SmallVector<int64_t, 4> newShape(newRank); in normalizeMemRefType() local
1788 newShape[d] = -1; in normalizeMemRefType()
1799 newShape[d] = ubConst.value() + 1; in normalizeMemRefType()
1806 .setShape(newShape) in normalizeMemRefType()
/llvm-project-15.0.7/mlir/lib/Dialect/Linalg/IR/
H A DLinalgOps.cpp1821 SmallVector<int64_t> newShape; in createNewOperandWithStaticSizes() local
1830 newShape.push_back(dimShape); in createNewOperandWithStaticSizes()
1836 newShape.push_back(affineExprToSize[dimExpr]); in createNewOperandWithStaticSizes()
1839 resultType = RankedTensorType::get(newShape, sourceType.getElementType()); in createNewOperandWithStaticSizes()
/llvm-project-15.0.7/mlir/lib/Conversion/TosaToLinalg/
H A DTosaToLinalg.cpp630 SmallVector<int64_t, 5> newShape; in elementwiseMatchAndRewriteHelper() local
632 newShape.reserve(type.getRank()); in elementwiseMatchAndRewriteHelper()
635 newShape.push_back(it.value()); in elementwiseMatchAndRewriteHelper()
641 if (newShape.size() != rank) { in elementwiseMatchAndRewriteHelper()
643 loc, RankedTensorType::get(newShape, type.getElementType()), operand, in elementwiseMatchAndRewriteHelper()
644 rewriter.getI64ArrayAttr(newShape)); in elementwiseMatchAndRewriteHelper()
/llvm-project-15.0.7/mlir/lib/Dialect/Vector/IR/
H A DVectorOps.cpp1609 SmallVector<int64_t, 4> newShape(type.getShape().begin(), in build() local
1614 newShape[dim.getPosition()] = newShape[dim.getPosition()] / multiplicity[i]; in build()
1616 VectorType resultType = VectorType::get(newShape, type.getElementType()); in build()