Home
last modified time | relevance | path

Searched defs:newShape (Results 1 – 23 of 23) sorted by relevance

/llvm-project/mlir/lib/Conversion/TosaToTensor/
H A DTosaToTensor.cpp32 ArrayRef<int64_t> newShape) { in inferReshapeInputType() argument
49 ArrayRef<int64_t> newShape) { in inferReshapeExpandedType() argument
238 auto newShape in matchAndRewrite() local
[all...]
/llvm-project/mlir/include/mlir/IR/
H A DBuiltinTypes.h212 setShape(ArrayRef<int64_t> newShape) setShape() argument
260 setShape(ArrayRef<int64_t> newShape) setShape() argument
/llvm-project/mlir/lib/Dialect/Tosa/Transforms/
H A DTosaDecomposeConv2D.cpp76 llvm::SmallVector<int64_t> newShape(inputType.getShape()); in matchAndRewrite() local
H A DTosaFolders.cpp319 llvm::SmallVector<int64_t> newShape(oldShape); in calculateReducedValue() local
380 auto newShape = resultType.getShape(); in matchAndRewrite() local
H A DTosaDecomposeDepthwise.cpp104 llvm::SmallVector<int64_t> newShape(inputType.getShape()); matchAndRewrite() local
/llvm-project/mlir/lib/Dialect/Tensor/Transforms/
H A DRewriteAsConstant.cpp96 auto newShape = in constantFoldPadOp() local
/llvm-project/mlir/lib/Dialect/Vector/Transforms/
H A DLowerVectorTransfer.cpp41 SmallVector<int64_t> newShape(addedRank, 1); in extendVectorRank() local
383 SmallVector<int64_t> newShape( matchAndRewriteMaskableOp() local
H A DVectorDropLeadUnitDim.cpp29 ArrayRef<int64_t> newShape = oldShape; in trimLeadingOneDims() local
H A DVectorTransferOpTransforms.cpp316 SmallVector<int64_t> newShape; trimNonScalableUnitDims() local
/llvm-project/mlir/unittests/IR/
H A DShapedTypeTest.cpp186 SmallVector<int64_t> newShape{2, 2}; TEST() local
/llvm-project/mlir/lib/Dialect/Affine/Transforms/
H A DPipelineDataTransfer.cpp85 SmallVector<int64_t, 4> newShape(1 + oldMemRefType.getRank()); in doubleBuffer() local
H A DLoopFusion.cpp265 SmallVector<int64_t, 4> newShape; in createPrivateMemRef() local
/llvm-project/mlir/lib/Dialect/Arith/Transforms/
H A DEmulateWideInt.cpp56 auto newShape = to_vector(type.getShape()); in reduceInnermostDim() local
121 auto newShape = llvm::to_vector(vecTy.getShape()); in appendX1Dim() local
1099 auto newShape = to_vector(ty.getShape()); in WideIntEmulationConverter() local
/llvm-project/flang/lib/Optimizer/Transforms/
H A DLoopVersioning.cpp463 fir::SequenceType::Shape newShape; runOnOperation() local
/llvm-project/mlir/lib/Dialect/Linalg/Transforms/
H A DSplitReduction.cpp79 SmallVector<int64_t> newShape; in splitReduction() local
H A DDropUnitDims.cpp613 SmallVector<int64_t> newShape; matchAndRewrite() local
/llvm-project/flang/lib/Optimizer/Dialect/
H A DFIRType.cpp1296 changeTypeShape(mlir::Type type,std::optional<fir::SequenceType::ShapeRef> newShape) changeTypeShape() argument
1321 std::optional<fir::SequenceType::ShapeRef> newShape; getBoxTypeWithNewShape() local
1328 std::optional<fir::SequenceType::ShapeRef> newShape; getBoxTypeWithNewShape() local
[all...]
/llvm-project/mlir/lib/Dialect/Bufferization/IR/
H A DBufferizationOps.cpp301 SmallVector<int64_t> newShape = llvm::to_vector(op.getType().getShape()); matchAndRewrite() local
/llvm-project/mlir/lib/Dialect/Tosa/IR/
H A DTosaCanonicalizations.cpp176 SmallVector<int64_t> newShape; in matchAndRewrite() local
/llvm-project/mlir/lib/Dialect/Affine/Utils/
H A DUtils.cpp1795 SmallVector<int64_t, 4> newShape(newRank); normalizeMemRefType() local
/llvm-project/mlir/lib/Dialect/Linalg/IR/
H A DLinalgOps.cpp2342 SmallVector<int64_t> newShape; createNewOperandWithStaticSizes() local
/llvm-project/mlir/lib/Dialect/Tensor/IR/
H A DTensorOps.cpp1720 SmallVector<int64_t, 4> newShape; inferCollapsedType() local
/llvm-project/mlir/lib/Dialect/Vector/IR/
H A DVectorOps.cpp5310 ArrayRef<int64_t> newShape = oldShape; trimTrailingOneDims() local