Searched refs:dynDims (Results 1 – 3 of 3) sorted by relevance
| /llvm-project-15.0.7/mlir/lib/Conversion/TosaToLinalg/ |
| H A D | TosaToLinalgNamed.cpp | 110 SmallVector<Value> dynDims; in inferDynamicDimsForConv() local
111 dynDims.resize(resultTy.getRank()); in inferDynamicDimsForConv()
114 dynDims[i] = rewriter.create<tensor::DimOp>(loc, input, i); in inferDynamicDimsForConv()
124 dynDims[heightDim] = getConvOutputDim( in inferDynamicDimsForConv()
137 dynDims[weightDim] = getConvOutputDim( in inferDynamicDimsForConv()
143 SmallVector<Value> filteredDims = condenseValues(dynDims); in inferDynamicDimsForConv()
511 SmallVector<Value> dynDims; in matchAndRewrite() local
526 SmallVector<Value> filteredDims = condenseValues(dynDims); in matchAndRewrite()
576 SmallVector<Value> dynDims; in matchAndRewrite() local
580 dynDims[0] = rewriter.create<tensor::DimOp>(loc, input, 0); in matchAndRewrite()
[all …]
|
| H A D | TosaToLinalg.cpp | 593 SmallVector<Value> dynDims; in elementwiseMatchAndRewriteHelper() local
599 if (operandTy.isDynamicDim(i) && !dynDims[i]) in elementwiseMatchAndRewriteHelper()
798 SmallVector<Value> dynDims; in reduceMatchAndRewriteHelper() local
1113 SmallVector<Value> dynDims; in matchAndRewrite() local
1660 SmallVector<Value> dynDims; in matchAndRewrite() local
1665 dynDims.push_back( in matchAndRewrite()
1721 SmallVector<Value> dynDims; in matchAndRewrite() local
1795 SmallVector<Value> dynDims; in matchAndRewrite() local
1942 SmallVector<Value> dynDims; in matchAndRewrite() local
2114 SmallVector<Value> dynDims; in matchAndRewrite() local
[all …]
|
| /llvm-project-15.0.7/mlir/lib/Dialect/Tensor/IR/ |
| H A D | TensorTilingInterfaceImpl.cpp | 220 SmallVector<Value> dynDims; in bubbleUpPadSlice() local
222 dispatchIndexOpFoldResults(sizes, dynDims, shape, ShapedType::kDynamicSize); in bubbleUpPadSlice()
237 loc, resultType, dynDims, in bubbleUpPadSlice()
|