Searched refs:dynamicDims (Results 1 – 8 of 8) sorted by relevance
| /llvm-project-15.0.7/mlir/include/mlir/Dialect/Tosa/Utils/ |
| H A D | CoversionUtils.h | 58 SmallVector<Value> dynamicDims; in checkHasDynamicBatchDims() local
66 return dynamicDims; in checkHasDynamicBatchDims()
76 dynamicDims.push_back( in checkHasDynamicBatchDims()
78 return dynamicDims; in checkHasDynamicBatchDims()
|
| /llvm-project-15.0.7/mlir/lib/Dialect/Tensor/Utils/ |
| H A D | Utils.cpp | 63 SmallVector<Value> dynamicDims; in createDynamicDimValues() local
66 dynamicDims.push_back( in createDynamicDimValues()
69 return dynamicDims; in createDynamicDimValues()
|
| /llvm-project-15.0.7/mlir/lib/Dialect/Bufferization/IR/ |
| H A D | BufferizationOps.cpp | 134 SmallVector<Value> &dynamicDims) { in populateDynamicDimSizes() argument
139 dynamicDims.push_back(b.create<memref::DimOp>(loc, shapedValue, i)); in populateDynamicDimSizes()
142 dynamicDims.push_back(b.create<tensor::DimOp>(loc, shapedValue, i)); in populateDynamicDimSizes()
190 SmallVector<Value> dynamicDims = getDynamicSizes(); in bufferize() local
192 assert(dynamicDims.empty() && "expected either `copy` or `dynamicDims`"); in bufferize()
193 populateDynamicDimSizes(rewriter, loc, copyBuffer, dynamicDims); in bufferize()
196 options.createAlloc(rewriter, loc, allocType, dynamicDims); in bufferize()
|
| /llvm-project-15.0.7/mlir/lib/Dialect/Tensor/IR/ |
| H A D | TensorInferTypeOpInterfaceImpl.cpp | 49 SmallVector<Value, 2> dynamicDims; in getCollapsedOutputDimFromInputShape() local
51 dynamicDims.push_back(builder.createOrFold<tensor::DimOp>(loc, src, dim)); in getCollapsedOutputDimFromInputShape()
57 dynamicDims)[0]; in getCollapsedOutputDimFromInputShape()
|
| /llvm-project-15.0.7/mlir/include/mlir/Dialect/Bufferization/IR/ |
| H A D | Bufferization.h | 39 SmallVector<Value> &dynamicDims);
|
| /llvm-project-15.0.7/mlir/lib/Conversion/TosaToLinalg/ |
| H A D | TosaToLinalgNamed.cpp | 699 SmallVector<Value> dynamicDims = dynamicDimsOr.value(); in matchAndRewrite() local
736 loc, dynamicDims, resultTy.getShape(), resultTy.getElementType()); in matchAndRewrite()
776 SmallVector<Value> dynamicDims = dynamicDimsOr.value(); in matchAndRewrite() local
798 loc, dynamicDims, accTy.getShape(), accETy); in matchAndRewrite()
823 loc, dynamicDims, resultTy.getShape(), resultETy); in matchAndRewrite()
|
| H A D | TosaToLinalg.cpp | 1169 SmallVector<Value> dynamicDims = dynamicDimsOr.value(); in matchAndRewrite() local
1245 loc, dynamicDims, outputTy.getShape(), outputTy.getElementType()); in matchAndRewrite()
1361 SmallVector<Value> dynamicDims = dynamicDimsOr.value(); in matchAndRewrite() local
1367 loc, dynamicDims, resultTy.getShape(), resultElementTy); in matchAndRewrite()
2056 SmallVector<Value> dynamicDims = dynamicDimsOr.value(); in matchAndRewrite() local
2064 .create<linalg::InitTensorOp>(loc, dynamicDims, resultTy.getShape(), in matchAndRewrite()
|
| /llvm-project-15.0.7/mlir/lib/Dialect/Linalg/Transforms/ |
| H A D | ElementwiseOpFusion.cpp | 1623 SmallVector<Value> dynamicDims; in matchAndRewrite() local
1627 dynamicDims.push_back(rewriter.createOrFold<tensor::DimOp>( in matchAndRewrite()
1631 loc, dynamicDims, operandType.getShape(), in matchAndRewrite()
|