| /llvm-project/mlir/lib/Dialect/Tosa/IR/ |
| H A D | TosaOps.cpp | 565 ShapeAdaptor inputShape(adaptor.getInput().getType()); inferReturnTypeComponents() local
590 ShapeAdaptor inputShape(adaptor.getInput().getType()); inferReturnTypeComponents() local
711 ShapeAdaptor inputShape(adaptor.getInput().getType()); inferReturnTypeComponents() local
768 ShapeAdaptor inputShape(adaptor.getInput1().getType()); inferReturnTypeComponents() local
855 ShapeAdaptor inputShape(adaptor.getInput().getType()); inferReturnTypeComponents() local
872 ShapeAdaptor inputShape(adaptor.getInput1().getType()); inferReturnTypeComponents() local
923 ShapeAdaptor inputShape(adaptor.getInput1().getType()); inferReturnTypeComponents() local
1009 ShapeAdaptor inputShape(adaptor.getInput1().getType()); inferReturnTypeComponents() local
1181 ShapeAdaptor inputShape(adaptor.getInput().getType()); inferReturnTypeComponents() local
1233 ShapeAdaptor inputShape(adaptor.getInput().getType()); inferReturnTypeComponents() local
1461 ShapeAdaptor inputShape(adaptor.getInput().getType()); inferReturnTypeComponents() local
1525 ShapeAdaptor inputShape(adaptor.getInput().getType()); inferReturnTypeComponents() local
1586 ShapeAdaptor inputShape(adaptor.getInput().getType()); inferReturnTypeComponents() local
1596 ShapeAdaptor inputShape(adaptor.getInput().getType()); inferReturnTypeComponents() local
1617 ShapeAdaptor inputShape(adaptor.getInput().getType()); inferReturnTypeComponents() local
1691 ShapeAdaptor inputShape(adaptor.getInput().getType()); inferReturnTypeComponents() local
[all...] |
| H A D | TosaCanonicalizations.cpp | 260 ArrayRef<int64_t> inputShape = inputType.getShape(); in matchAndRewrite() local
|
| /llvm-project/mlir/lib/Dialect/Tosa/Transforms/ |
| H A D | TosaDecomposeConv2D.cpp | 100 llvm::SmallVector<int64_t, 2> revisedInputShape{combined, inputShape[3]}; in matchAndRewrite() local
|
| H A D | TosaDecomposeDepthwise.cpp | 51 ArrayRef<int64_t> inputShape = inputType.getShape(); matchAndRewrite() local
|
| H A D | TosaFolders.cpp | 145 auto inputShape = inputType.getShape(); in transposeType() local
|
| /llvm-project/mlir/lib/Dialect/Tensor/Transforms/ |
| H A D | RewriteAsConstant.cpp | 59 int64_t transformIndexSpace(ArrayRef<int64_t> inputShape, in transformIndexSpace()
|
| /llvm-project/mlir/examples/toy/Ch3/mlir/ |
| H A D | Dialect.cpp | 308 auto inputShape = inputType.getShape(); in verify() local
|
| /llvm-project/mlir/examples/toy/Ch2/mlir/ |
| H A D | Dialect.cpp | 308 auto inputShape = inputType.getShape(); in verify() local
|
| /llvm-project/mlir/examples/toy/Ch4/mlir/ |
| H A D | Dialect.cpp | 429 auto inputShape = inputType.getShape(); in verify() local
|
| /llvm-project/mlir/examples/toy/Ch6/mlir/ |
| H A D | Dialect.cpp | 429 auto inputShape = inputType.getShape(); in verify() local
|
| /llvm-project/mlir/examples/toy/Ch5/mlir/ |
| H A D | Dialect.cpp | 429 auto inputShape = inputType.getShape(); in verify() local
|
| /llvm-project/mlir/lib/Dialect/Arith/Utils/ |
| H A D | Utils.cpp | 27 ArrayRef<OpFoldResult> inputShape) { in inferExpandShapeOutputShape() argument
|
| /llvm-project/mlir/lib/Dialect/Linalg/IR/ |
| H A D | LinalgOps.cpp | 1968 auto inputShape = inputType.getShape(); verify() local
2450 ArrayRef<int64_t> inputShape = inputType.getShape(); verify() local
2616 ArrayRef<int64_t> inputShape = inputType.getShape(); reduce() local
2639 ArrayRef<int64_t> inputShape = inputType.getShape(); buildSubAndExpOp() local
2665 ArrayRef<int64_t> inputShape = inputType.getShape(); buildDivOp() local
[all...] |
| /llvm-project/mlir/examples/toy/Ch7/mlir/ |
| H A D | Dialect.cpp | 491 auto inputShape = inputType.getShape(); in verify() local
|
| /llvm-project/mlir/lib/Dialect/Linalg/Transforms/ |
| H A D | ConvertConv2DToImg2Col.cpp | 238 ArrayRef<int64_t> inputShape = operandTensorType.getShape(); in rewriteInIm2Col() local
|
| H A D | Vectorization.cpp | 922 auto inputShape = cast<ShapedType>(extractOp.getTensor().getType()); getTensorExtractMemoryAccessPattern() local
1522 SmallVector<int64_t> inputShape(inputVectorSizes); vectorizeAsTensorPackOp() local
|
| H A D | Transforms.cpp | 1146 auto inputShape = packOp.getSourceType().getShape(); matchAndRewrite() local
|
| /llvm-project/mlir/lib/Conversion/TosaToLinalg/ |
| H A D | TosaToLinalgNamed.cpp | 45 auto inputShape = inputTy.getShape(); in applyPad() local
|
| H A D | TosaToLinalg.cpp | 1884 auto inputShape = inputTy.getShape(); matchAndRewrite() local
|
| /llvm-project/mlir/lib/Dialect/Tensor/IR/ |
| H A D | TensorOps.cpp | 1655 inferOutputShape(OpBuilder & b,Location loc,RankedTensorType expandedType,ArrayRef<ReassociationIndices> reassociation,ArrayRef<OpFoldResult> inputShape) inferOutputShape() argument
1678 SmallVector<OpFoldResult> inputShape = build() local
3896 requirePaddingValue(ArrayRef<int64_t> inputShape,ArrayRef<int64_t> innerDimsPos,ArrayRef<int64_t> outputShape,ArrayRef<int64_t> outerDimsPerm,ArrayRef<OpFoldResult> innerTiles) requirePaddingValue() argument
[all...] |
| /llvm-project/mlir/lib/Dialect/MemRef/IR/ |
| H A D | MemRefOps.cpp | 2269 inferOutputShape(OpBuilder & b,Location loc,MemRefType expandedType,ArrayRef<ReassociationIndices> reassociation,ArrayRef<OpFoldResult> inputShape) inferOutputShape() argument
2292 SmallVector<OpFoldResult> inputShape = build() local
|
| /llvm-project/mlir/lib/Dialect/Math/Transforms/ |
| H A D | PolynomialApproximation.cpp | 110 if (inputShape == llvm::ArrayRef(vectorWidth)) in handleMultidimensionalVectors() local
|