| /llvm-project/mlir/lib/Dialect/Tosa/IR/ |
| H A D | TosaOps.cpp | 390 buildTransConvOpWithQuantInfo(OpBuilder & builder,OperationState & result,Type outputType,Value input,Value weight,Value bias,DenseI64ArrayAttr outpad,DenseI64ArrayAttr stride,DenseI64ArrayAttr outputShape) buildTransConvOpWithQuantInfo() argument
595 llvm::SmallVector<int64_t> outputShape; inferReturnTypeComponents() local
630 llvm::SmallVector<int64_t> outputShape; inferReturnTypeComponents() local
770 SmallVector<int64_t> outputShape; inferReturnTypeComponents() local
873 SmallVector<int64_t> outputShape; inferReturnTypeComponents() local
1030 SmallVector<int64_t> outputShape; inferReturnTypeComponents() local
1153 llvm::SmallVector<int64_t> outputShape; inferReturnTypeComponents() local
1178 llvm::SmallVector<int64_t, 4> outputShape; inferReturnTypeComponents() local
1217 llvm::SmallVector<int64_t> outputShape; inferReturnTypeComponents() local
1254 SmallVector<int64_t> outputShape; ReduceInferReturnTypes() local
1332 auto outputShape = outputType.getShape(); REDUCE_SHAPE_INFER() local
1418 llvm::SmallVector<int64_t> outputShape; NARY_SHAPE_INFER() local
1452 llvm::SmallVector<int64_t> outputShape(4, ShapedType::kDynamic); inferReturnTypeComponents() local
1514 llvm::SmallVector<int64_t> outputShape(5, ShapedType::kDynamic); inferReturnTypeComponents() local
1606 llvm::SmallVector<int64_t> outputShape(4, ShapedType::kDynamic); inferReturnTypeComponents() local
1682 llvm::SmallVector<int64_t> outputShape = inferReturnTypeComponents() local
[all...] |
| H A D | TosaCanonicalizations.cpp | 255 ArrayRef<int64_t> outputShape = outputType.getShape(); in matchAndRewrite() local
|
| /llvm-project/mlir/lib/Dialect/Tosa/Transforms/ |
| H A D | TosaDecomposeConv2D.cpp | 149 rewriter.getDenseI64ArrayAttr(convertFromMlirShape(outputShape))); in matchAndRewrite() local
|
| H A D | TosaDecomposeDepthwise.cpp | 144 auto outputShape = cast<ShapedType>(op.getOutput().getType()).getShape(); matchAndRewrite() local
|
| /llvm-project/mlir/lib/Dialect/Linalg/Transforms/ |
| H A D | ConvertConv2DToImg2Col.cpp | 103 ArrayRef<int64_t> outputShape = outputType.getShape(); in rewriteInIm2Col() local
279 ArrayRef<int64_t> outputShape = outputType.getShape(); in rewriteInIm2Col() local
388 auto outputShape = outputType.getShape(); in rewriteInIm2Col() local
522 ArrayRef<int64_t> outputShape = outputType.getShape(); in rewriteInIm2Col() local
|
| H A D | ConstantFold.cpp | 160 auto outputShape = outputType.getShape(); in matchAndRewrite() local
|
| H A D | Transforms.cpp | 1334 auto outputShape = outputType.getShape(); returningMatchAndRewrite() local
1458 auto outputShape = outputType.getShape(); returningMatchAndRewrite() local
1526 auto outputShape = outputType.getShape(); returningMatchAndRewrite() local
|
| /llvm-project/mlir/test/lib/Dialect/Tensor/ |
| H A D | TestTensorTransforms.cpp | 194 SmallVector<OpFoldResult> outputShape = reifiedShapes[0]; matchAndRewrite() local
|
| /llvm-project/mlir/lib/Dialect/MemRef/IR/ |
| H A D | MemRefOps.cpp | 2270 std::optional<SmallVector<OpFoldResult>> outputShape = inferOutputShape() local
2281 build(OpBuilder & builder,OperationState & result,Type resultType,Value src,ArrayRef<ReassociationIndices> reassociation,ArrayRef<OpFoldResult> outputShape) build() argument
2295 FailureOr<SmallVector<OpFoldResult>> outputShape = inferOutputShape( build() local
2319 build(OpBuilder & builder,OperationState & result,ArrayRef<int64_t> resultShape,Value src,ArrayRef<ReassociationIndices> reassociation,ArrayRef<OpFoldResult> outputShape) build() argument
[all...] |
| /llvm-project/mlir/lib/Dialect/MemRef/Transforms/ |
| H A D | FoldMemRefAliasOps.cpp | 87 auto outputShape = inferExpandShapeOutputShape( in resolveSourceIndicesExpandShape() local
|
| /llvm-project/mlir/lib/Dialect/Tensor/IR/ |
| H A D | TensorTilingInterfaceImpl.cpp | 215 ReifiedRankedShapedTypeDims outputShape; getResultTilePosition() local
|
| H A D | TensorOps.cpp | 1656 std::optional<SmallVector<OpFoldResult>> outputShape = inferOutputShape() local
1667 build(OpBuilder & builder,OperationState & result,Type resultType,Value src,ArrayRef<ReassociationIndices> reassociation,ArrayRef<OpFoldResult> outputShape) build() argument
1681 FailureOr<SmallVector<OpFoldResult>> outputShape = inferOutputShape( build() local
3898 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/Linalg/IR/ |
| H A D | LinalgOps.cpp | 1474 auto outputShape = getInit().getType().getShape(); verify() local
2451 ArrayRef<int64_t> outputShape = outputType.getShape(); verify() local
[all...] |