Home
last modified time | relevance | path

Searched refs:dilation (Results 1 – 16 of 16) sorted by relevance

/llvm-project-15.0.7/mlir/lib/Dialect/Linalg/Transforms/
H A DNamedOpConversions.cpp32 Attribute dilation, PatternRewriter &rewriter) { in matchAndReplaceDepthwiseConv() argument
75 ValueRange{collapsedInit}, stride, dilation) in matchAndReplaceDepthwiseConv()
82 ValueRange{collapsedInit}, stride, dilation) in matchAndReplaceDepthwiseConv()
108 auto dilation = op.dilations(); in matchAndRewrite() local
111 nullptr, init, stride, dilation, in matchAndRewrite()
130 auto dilation = op.dilations(); in matchAndRewrite() local
133 init, stride, dilation, rewriter); in matchAndRewrite()
H A DVectorization.cpp1665 auto dilation = dilations ? *dilations.getValues<uint64_t>().begin() : 1; in vectorizeConvolution() local
1666 Conv1DNwcGenerator e(b, op, stride, dilation); in vectorizeConvolution()
/llvm-project-15.0.7/mlir/test/Dialect/Tosa/
H A Dinvalid.mlir6 %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]}
34 %0 = "tosa.conv2d"(%arg0, %arg1, %arg2) {dilation = [1, 1], pad = [0, 0, 0, 0], stride = [1, 1]}
H A Dtosa-decompose-transpose-conv.mlir7 …// 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 Dtosa-decompose-conv2d.mlir17 …%0 = "tosa.conv2d"(%arg0, %arg1, %arg2) {pad = [0, 0, 0, 0], stride = [1, 1], dilation = [1, 1]} :…
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 Dtosa-decompose-depthwise.mlir19 …%0 = "tosa.depthwise_conv2d"(%arg0, %arg1, %arg2) {pad = [0, 0, 0, 0], stride = [1, 1], dilation =…
28 …%0 = "tosa.depthwise_conv2d"(%arg0, %arg1, %arg2) {pad = [0, 0, 0, 0], stride = [1, 1], dilation =…
H A Dtosa-infer-shapes.mlir690 …%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…
852 …%0 = "tosa.depthwise_conv2d"(%arg0, %arg1, %arg2) {dilation = [1, 1], pad = [0, 0, 0, 0], stride =…
861 …%0 = "tosa.depthwise_conv2d"(%arg0, %arg1, %arg2) {dilation = [1, 1], pad = [0, 0, 0, 0], stride =…
870 …%0 = "tosa.depthwise_conv2d"(%arg0, %arg1, %arg2) {dilation = [1, 1], pad = [0, 0, 0, 0], stride =…
[all …]
H A Dquant-test.mlir15 …%0 = "tosa.conv2d"(%arg0, %arg1, %arg2) {pad = [1, 1, 2, 2], dilation = [2, 1], stride = [1, 1], q…
H A Dcanonicalize.mlir109 …%0 = "tosa.conv2d"(%arg0, %weight, %bias) {pad = [0, 0, 0, 0], stride = [2, 2], dilation = [1, 1]}…
118 …%0 = "tosa.conv2d"(%arg0, %weight, %bias) {pad = [0, 0, 0, 0], stride = [1, 1], dilation = [1, 1]}…
125 …%0 = "tosa.depthwise_conv2d"(%arg0, %arg1, %arg2) {pad = [0, 0, 0, 0], stride = [2, 2], dilation =…
132 …%0 = "tosa.depthwise_conv2d"(%arg0, %arg1, %arg2) {pad = [0, 0, 0, 0], stride = [1, 1], dilation =…
140 …%0 = "tosa.max_pool2d"(%arg0) {kernel = [1, 1], pad = [0, 0, 0, 0], stride = [1, 1], dilation = [1…
H A Dops.mlir43 …%0 = "tosa.conv2d"(%arg0, %arg1, %arg2) {dilation = [1, 1], pad = [0, 0, 0, 0], stride = [1, 1]} :…
50 …%2 = "tosa.depthwise_conv2d"(%arg0, %arg1, %arg2) {dilation = [1, 1], pad = [0, 0, 0, 0], stride =…
/llvm-project-15.0.7/mlir/lib/Dialect/Tosa/IR/
H A DTosaOps.cpp180 result.addAttribute("dilation", dilation); in buildConvOpWithQuantInfo()
1050 llvm::SmallVector<int64_t> dilation; in inferReturnTypeComponents() local
1054 getI64Values(adaptor.getDilation(), dilation); in inferReturnTypeComponents()
1069 int32_t filterSize = (weightWidth - 1) * dilation[1] + 1; in inferReturnTypeComponents()
1120 llvm::SmallVector<int64_t> dilation; in inferReturnTypeComponents() local
1124 getI64Values(adaptor.getDilation(), dilation); in inferReturnTypeComponents()
1139 int32_t filterSize = (weightWidth - 1) * dilation[1] + 1; in inferReturnTypeComponents()
1147 int32_t filterSize = (weightDepth - 1) * dilation[2] + 1; in inferReturnTypeComponents()
1222 llvm::SmallVector<int64_t> dilation; in inferReturnTypeComponents() local
1226 getI64Values(adaptor.getDilation(), dilation); in inferReturnTypeComponents()
[all …]
/llvm-project-15.0.7/mlir/test/Conversion/TosaToLinalg/
H A Dtosa-to-linalg-named.mlir357 …%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 …
486 …%2 = "tosa.depthwise_conv2d"(%arg0, %arg1, %arg2) { pad = [0, 0, 0, 0], stride = [1, 1], dilation …
510 …%2 = "tosa.depthwise_conv2d"(%arg0, %arg1, %arg2) { pad = [0, 0, 0, 0], stride = [1, 1], dilation …
532 …%2 = "tosa.depthwise_conv2d"(%arg0, %arg1, %arg2) { pad = [0, 0, 0, 0], stride = [2, 2], dilation …
560 …o = #tosa.conv_quant<input_zp = -128, weight_zp = 42>, stride = [1, 1], dilation = [1, 1] } : (ten…
584 …o = #tosa.conv_quant<input_zp = -128, weight_zp = 42>, stride = [1, 1], dilation = [2, 2] } : (ten…
[all …]
/llvm-project-15.0.7/mlir/lib/Conversion/TosaToLinalg/
H A DTosaToLinalgNamed.cpp89 Value dilation = reifyConstantDim(dilationAttr, builder); in getConvOutputDim() local
90 Value dilated = builder.create<arith::MulIOp>(dilation, subOne); in getConvOutputDim()
253 llvm::SmallVector<int64_t> stride, dilation; in matchAndRewrite() local
255 getValuesFromIntArrayAttribute(dilationTosaAttr, dilation); in matchAndRewrite()
261 RankedTensorType::get({2}, rewriter.getI64Type()), dilation); in matchAndRewrite()
406 llvm::SmallVector<int64_t> stride, dilation; in matchAndRewrite() local
408 getValuesFromIntArrayAttribute(dilationTosaAttr, dilation); in matchAndRewrite()
414 RankedTensorType::get({2}, rewriter.getI64Type()), dilation); in matchAndRewrite()
/llvm-project-15.0.7/mlir/include/mlir/Dialect/Tosa/IR/
H A DTosaOpBase.td130 "ArrayAttr":$pad, "ArrayAttr":$stride, "ArrayAttr":$dilation),
134 pad, stride, dilation);
149 // strides/dilation/padding.
H A DTosaOps.td109 Tosa_IntArrayAttr2:$dilation,
141 Tosa_IntArrayAttr3:$dilation,
174 Tosa_IntArrayAttr2:$dilation,
/llvm-project-15.0.7/mlir/test/Interfaces/TilingInterface/
H A Dtile-using-interface.mlir143 dilation = dense<[4, 5]> : tensor<2xi64>,
190 // CHECK-SAME: dilation = dense<[4, 5]> : tensor<2xi64>, strides = dense<[2, 3]> : tens…