| /llvm-project-15.0.7/mlir/lib/Dialect/Tosa/Transforms/ |
| H A D | TosaDecomposeTransposeConv.cpp | 128 Value conv2d; in matchAndRewrite() local
130 conv2d = rewriter.create<tosa::Conv2DOp>( in matchAndRewrite()
135 conv2d = rewriter.create<tosa::Conv2DOp>( in matchAndRewrite()
141 rewriter.replaceOp(op, conv2d); in matchAndRewrite()
290 Value conv2d; in matchAndRewrite() local
292 conv2d = createOpAndInfer<tosa::Conv2DOp>( in matchAndRewrite()
301 conv2d = createOpAndInfer<tosa::Conv2DOp>( in matchAndRewrite()
311 ShapedType convTy = conv2d.getType().cast<ShapedType>(); in matchAndRewrite()
320 conv2d = createOpAndInfer<tosa::ReshapeOp>( in matchAndRewrite()
329 conv2d = createOpAndInfer<tosa::TransposeOp>( in matchAndRewrite()
[all …]
|
| /llvm-project-15.0.7/mlir/test/Dialect/Tosa/ |
| H A D | invalid.mlir | 6 %0 = "tosa.conv2d"(%arg0, %arg1, %arg2) {dilation = [1, 1], pad = [0, 0, 0, 0], stride = [1, 1]}
15 %0 = "tosa.conv2d"(%arg0, %arg1, %arg2) {dilation = [1, 1], pad = [0, 0, 0, 0], stride = [1, 1]}
24 %0 = "tosa.conv2d"(%arg0, %arg1, %arg2) {dilation = [1, 1], pad = [0, 0, 0, 0], stride = [1, 1]}
33 …// expected-error@+1 {{'tosa.conv2d' op quantizationattr is required for quantized type, and not a…
34 %0 = "tosa.conv2d"(%arg0, %arg1, %arg2) {dilation = [1, 1], pad = [0, 0, 0, 0], stride = [1, 1]}
|
| H A D | tosa-decompose-conv2d.mlir | 7 // CHECK-NOT: "tosa.conv2d"
17 …%0 = "tosa.conv2d"(%arg0, %arg1, %arg2) {pad = [0, 0, 0, 0], stride = [1, 1], dilation = [1, 1]} :…
25 // CHECK-NOT: "tosa.conv2d"
36 …%0 = "tosa.conv2d"(%arg0, %arg1, %arg2) {pad = [0, 0, 0, 0], stride = [1, 1], dilation = [1, 1], q…
53 …%0 = "tosa.conv2d"(%arg0, %arg1, %arg2) {dilation = [1, 1], pad = [0, 0, 0, 0], quantization_info …
|
| H A D | tosa-decompose-transpose-conv.mlir | 7 …// CHECK: "tosa.conv2d"(%arg0, %[[REV2]], %arg2) {dilation = [1, 1], pad = [2, 2, 5, 5], stride = …
19 …// CHECK: "tosa.conv2d"(%arg0, %[[REV2]], %arg2) {dilation = [1, 1], pad = [2, 2, 5, 5], quantizat…
30 …// CHECK: "tosa.conv2d"(%arg0, %[[REV2]], %arg2) {dilation = [1, 1], pad = [1, 1, 2, 2], quantizat…
56 …// CHECK-DAG: %[[CONV:.+]] = "tosa.conv2d"(%[[NEWINPUT]], %[[NEWWEIGHT]], %[[BIAS]]) {dilation = […
88 …// CHECK-DAG: %[[CONV:.+]] = "tosa.conv2d"(%[[NEWINPUT]], %[[NEWWEIGHT]], %[[BIAS]]) {dilation = […
|
| H A D | quant-test.mlir | 14 // CHECK: tosa.conv2d
15 …%0 = "tosa.conv2d"(%arg0, %arg1, %arg2) {pad = [1, 1, 2, 2], dilation = [2, 1], stride = [1, 1], q…
|
| H A D | canonicalize.mlir | 106 // CHECK: "tosa.conv2d"
109 …%0 = "tosa.conv2d"(%arg0, %weight, %bias) {pad = [0, 0, 0, 0], stride = [2, 2], dilation = [1, 1]}…
115 // CHECK: "tosa.conv2d"
118 …%0 = "tosa.conv2d"(%arg0, %weight, %bias) {pad = [0, 0, 0, 0], stride = [1, 1], dilation = [1, 1]}…
|
| H A D | tosa-infer-shapes.mlir | 690 …%0 = "tosa.conv2d"(%input, %weights, %bias) {pad = [0, 0, 0, 0], stride = [1, 1], dilation = [1, 1…
699 …%0 = "tosa.conv2d"(%input, %weights, %bias) {pad = [0, 0, 0, 0], stride = [1, 1], dilation = [1, 1…
732 …%0 = "tosa.conv2d"(%input, %weights, %bias) {pad = [0, 0, 0, 0], stride = [1, 1], dilation = [1, 1…
741 …%0 = "tosa.conv2d"(%input, %weights, %bias) {pad = [0, 0, 0, 0], stride = [1, 1], dilation = [1, 1…
762 …%0 = "tosa.conv2d"(%input, %weights, %bias) {pad = [1, 2, 3, 4], stride = [1, 1], dilation = [1, 1…
771 …%0 = "tosa.conv2d"(%input, %weights, %bias) {pad = [0, 0, 0, 0], stride = [1, 1], dilation = [3, 2…
780 …%0 = "tosa.conv2d"(%input, %weights, %bias) {pad = [0, 0, 0, 0], stride = [3, 2], dilation = [1, 1…
|
| H A D | ops.mlir | 41 // CHECK-LABEL: conv2d
43 …%0 = "tosa.conv2d"(%arg0, %arg1, %arg2) {dilation = [1, 1], pad = [0, 0, 0, 0], stride = [1, 1]} :…
|
| /llvm-project-15.0.7/mlir/test/Integration/Dialect/SparseTensor/CPU/ |
| H A D | sparse_filter_conv2d.mlir | 18 func.func @conv2d(%input: tensor<8x8xi32>,
53 %0 = call @conv2d(%input, %sparse_filter, %output)
|
| /llvm-project-15.0.7/mlir/test/Conversion/TosaToLinalg/ |
| H A D | tosa-to-linalg-named.mlir | 357 …%0 = "tosa.conv2d"(%input, %weights, %bias) {pad = [0, 0, 0, 0], stride = [1, 1], dilation = [2, 1…
380 …%0 = "tosa.conv2d"(%input, %weights, %bias) {pad = [0, 0, 0, 0], stride = [1, 1], dilation = [2, 1…
440 …%0 = "tosa.conv2d"(%input, %weights, %bias) {pad = [0, 0, 0, 0], stride = [1, 1], dilation = [2, 1…
452 …%0 = "tosa.conv2d"(%input, %weights, %bias) {pad = [1, 1, 1, 1], stride = [1, 1], dilation = [2, 1…
464 …%0 = "tosa.conv2d"(%arg0, %arg1, %arg2) {dilation = [1, 1], pad = [1, 1, 1, 1], quantization_info …
|
| /llvm-project-15.0.7/mlir/test/Dialect/SparseTensor/ |
| H A D | sparse_kernels.mlir | 154 // CHECK-LABEL: func @conv2d(
194 func.func @conv2d(%input: tensor<8x8xi32>,
|
| /llvm-project-15.0.7/mlir/include/mlir/Dialect/Tosa/IR/ |
| H A D | TosaOps.td | 89 // Operator: conv2d
91 def Tosa_Conv2DOp : Tosa_Op<"conv2d", [
|
| /llvm-project-15.0.7/mlir/docs/Rationale/ |
| H A D | Rationale.md | 661 func.func @conv2d(%input: memref<16x1024x1024x3xf32, #lm0, /*scratchpad=*/1>,
|