Home
last modified time | relevance | path

Searched refs:dilations (Results 1 – 22 of 22) sorted by relevance

/llvm-project-15.0.7/mlir/test/Dialect/Linalg/
H A Dnamed-ops.mlir157 // CHECK-NOT: dilations =
217 // CHECK-SAME: dilations = dense<1> : tensor<1xi64>
233 // CHECK-SAME: dilations = dense<1> : tensor<1xi64>
249 // CHECK-SAME: dilations = dense<1> : tensor<2xi64>
265 // CHECK-SAME: dilations = dense<1> : tensor<2xi64>
281 // CHECK-SAME: dilations = dense<1> : tensor<2xi64>
297 // CHECK-SAME: dilations = dense<1> : tensor<2xi64>
313 // CHECK-SAME: dilations = dense<1> : tensor<2xi64>
329 // CHECK-SAME: dilations = dense<1> : tensor<2xi64>
345 // CHECK-SAME: dilations = dense<1> : tensor<3xi64>
[all …]
H A Dnamedop_conversion.mlir7 …// CHECK-DAG: %[[CONV:.+]] = linalg.depthwise_conv_2d_nhwc_hwc {dilations = dense<1> : tensor<2xi6…
9 …%0 = linalg.depthwise_conv_2d_nhwc_hwcm {dilations = dense<1> : tensor<2xi64>, strides = dense<2> …
20 …// CHECK-DAG: %[[CONV:.+]] = linalg.depthwise_conv_2d_nhwc_hwc_q {dilations = dense<1> : tensor<2x…
22 …%0 = linalg.depthwise_conv_2d_nhwc_hwcm_q {dilations = dense<1> : tensor<2xi64>, strides = dense<2…
H A Dgeneralize-named-polymorphic-ops.mlir121 …%0 = linalg.pooling_nhwc_max {dilations = dense<[1, 2]> : tensor<2xi64>, strides = dense<[2, 4]> :…
135 …%0 = linalg.pooling_nhwc_max {dilations = dense<[1, 2]> : tensor<2xi64>, strides = dense<[2, 4]> :…
147 …%0 = linalg.pooling_nhwc_max_unsigned {dilations = dense<[1, 2]> : tensor<2xi64>, strides = dense<…
159 …%0 = linalg.pooling_nhwc_min {dilations = dense<[1, 2]> : tensor<2xi64>, strides = dense<[2, 4]> :…
173 …%0 = linalg.pooling_nhwc_min {dilations = dense<[1, 2]> : tensor<2xi64>, strides = dense<[2, 4]> :…
185 …%0 = linalg.pooling_nhwc_min_unsigned {dilations = dense<[1, 2]> : tensor<2xi64>, strides = dense<…
197 …%0 = linalg.pooling_nhwc_sum {dilations = dense<[1, 2]> : tensor<2xi64>, strides = dense<[2, 4]> :…
211 …%0 = linalg.pooling_nhwc_sum {dilations = dense<[1, 2]> : tensor<2xi64>, strides = dense<[2, 4]> :…
H A Dtransform-op-decompose.mlir13 %0 = linalg.conv_2d_nhwc_hwcf {dilations = dense<1> : tensor<2xi64>,
34 …%0 = linalg.depthwise_conv_2d_nhwc_hwc {dilations = dense<1> : vector<2xi64>, strides = dense<2> :…
H A Dvectorize-convolution.mlir5 {dilations = dense<1> : tensor<1xi64>, strides = dense<3> : tensor<1xi64>}
66 {dilations = dense<2> : tensor<1xi64>, strides = dense<3> : tensor<1xi64>}
143 {dilations = dense<2> : tensor<1xi64>, strides = dense<1> : tensor<1xi64>}
192 {dilations = dense<2> : tensor<1xi64>, strides = dense<1> : tensor<1xi64>}
233 {dilations = dense<1> : vector<1xi64>, strides = dense<1> : vector<1xi64>}
H A Dgeneralize-named-ops.mlir77 { dilations = dense<1> : tensor<2xi64>, strides = dense<1> : tensor<2xi64> }
104 { dilations = dense<2> : tensor<2xi64>, strides = dense<1> : tensor<2xi64> }
130 …linalg.depthwise_conv_2d_nhwc_hwc {dilations = dense<1> : vector<2xi64>, strides = dense<2> : vect…
156 linalg.conv_1d_nwc_wcf {dilations = dense<1> : tensor<1xi64>,
H A Dbubble-up-extract-slice-op.mlir142 {dilations = dense<1> : tensor<2xi64>, strides = dense<2> : tensor<2xi64>}
157 // CHECK: %[[CONV:.+]] = linalg.conv_2d_nhwc_hwcf {dilations = dense<1> : tensor<2xi64>, strides = …
H A Dtile-and-fuse-tensors.mlir72 {dilations = dense<1> : tensor<2xi64>, strides = dense<2> : tensor<2xi64>}
154 {dilations = dense<1> : tensor<2xi64>, strides = dense<2> : tensor<2xi64>}
H A Dinvalid.mlir409 { dilations = dense<1> : tensor<2xi64>, strides = dense<1> : tensor<2xi64>}
/llvm-project-15.0.7/mlir/test/python/dialects/linalg/opdsl/
H A Demit_pooling.py22 dilations=IndexAttrDef(S.DH, S.DW, default=[1, 1])): argument
55 input, shape, outs=[init_result], strides=[2, 4], dilations=[1, 2])
72 dilations=[1, 2])
88 input, shape, outs=[init_result], strides=[2, 4], dilations=[1, 2])
104 dilations=[1, 2])
121 dilations=[1, 2])
136 dilations=[1, 2])
H A Demit_convolution.py20 dilations=IndexAttrDef(S.DH, S.DW, default=[1, 2])): argument
/llvm-project-15.0.7/mlir/python/mlir/dialects/linalg/opdsl/ops/
H A Dcore_named_ops.py260 dilations=IndexAttrDef(S.DW, default=[1])): argument
401 dilations=IndexAttrDef(S.DD, argument
469 dilations=IndexAttrDef(S.DH, argument
496 dilations=IndexAttrDef(S.DH, argument
526 dilations=IndexAttrDef(S.DH, argument
556 dilations=IndexAttrDef(S.DH, argument
588 dilations=IndexAttrDef(S.DH, argument
617 dilations=IndexAttrDef(S.DD, argument
819 dilations=IndexAttrDef(S.DD, argument
846 dilations=IndexAttrDef(S.DD, argument
[all …]
/llvm-project-15.0.7/mlir/lib/Dialect/Linalg/Transforms/
H A DTransforms.cpp901 auto dilations = llvm::to_vector<4>(convOp.dilations().getValues<int64_t>()); in returningMatchAndRewrite() local
902 dilations.erase(dilations.begin() + (removeH ? 0 : 1)); in returningMatchAndRewrite()
903 auto dilationsAttr = rewriter.getI64VectorAttr(dilations); in returningMatchAndRewrite()
971 auto dilations = llvm::to_vector<4>(convOp.dilations().getValues<int64_t>()); in returningMatchAndRewrite() local
972 dilations.erase(dilations.begin() + (removeH ? 0 : 1)); in returningMatchAndRewrite()
973 auto dilationsAttr = rewriter.getI64VectorAttr(dilations); in returningMatchAndRewrite()
H A DNamedOpConversions.cpp108 auto dilation = op.dilations(); in matchAndRewrite()
130 auto dilation = op.dilations(); in matchAndRewrite()
H A DVectorization.cpp1663 auto dilations = op->getAttrOfType<DenseIntElementsAttr>("dilations"); in vectorizeConvolution() local
1665 auto dilation = dilations ? *dilations.getValues<uint64_t>().begin() : 1; in vectorizeConvolution()
/llvm-project-15.0.7/mlir/test/Conversion/TosaToLinalg/
H A Dtosa-to-linalg-named.mlir148 …// CHECK: linalg.pooling_nhwc_max {dilations = dense<1> : vector<2xi64>, strides = dense<1> : vect…
162 …// CHECK: linalg.pooling_nhwc_max {dilations = dense<1> : vector<2xi64>, strides = dense<1> : vect…
175 …// CHECK: linalg.pooling_nhwc_max {dilations = dense<1> : vector<2xi64>, strides = dense<1> : vect…
214 …// CHECK: [[POOL:%.+]] = linalg.pooling_nhwc_sum {dilations = dense<1> : vector<2xi64>, strides = …
275 …// CHECK: %[[POOL:.+]] = linalg.pooling_nhwc_sum {dilations = dense<1> : vector<2xi64>, strides = …
353 …// CHECK: %[[CONV:.+]] = linalg.conv_2d_nhwc_hwcf {dilations = dense<[2, 1]> : tensor<2xi64>, stri…
376 …// CHECK: %[[CONV:.+]] = linalg.conv_2d_nhwc_hwcf {dilations = dense<[2, 1]> : tensor<2xi64>, stri…
436 …// CHECK: %[[CONV:.+]] = linalg.conv_2d_nhwc_hwcf {dilations = dense<[2, 1]> : tensor<2xi64>, stri…
479 …// CHECK: [[DEPTH:%.+]] = linalg.depthwise_conv_2d_nhwc_hwcm {dilations = dense<1> : tensor<2xi64>…
503 …// CHECK: %[[DEPTH:.+]] = linalg.depthwise_conv_2d_nhwc_hwcm {dilations = dense<1> : tensor<2xi64>…
[all …]
/llvm-project-15.0.7/mlir/include/mlir/Dialect/Linalg/IR/
H A DLinalgNamedStructuredOps.yaml1167 name: dilations
1259 name: dilations
1357 name: dilations
1464 name: dilations
1592 name: dilations
1690 name: dilations
1787 name: dilations
1881 name: dilations
1964 name: dilations
2053 name: dilations
[all …]
/llvm-project-15.0.7/mlir/test/python/integration/dialects/linalg/
H A Dopsrun.py494 input, shape, outs=[output], strides=[2, 4], dilations=[1, 2])
529 dilations=[1, 2],
/llvm-project-15.0.7/mlir/test/Integration/Dialect/Linalg/CPU/
H A Dtest-conv-1d-nwc-wcf-call.mlir22 linalg.conv_1d_nwc_wcf {dilations = dense<1> : tensor<1xi64>,
H A Dtest-conv-2d-nhwc-hwcf-call.mlir22 linalg.conv_2d_nhwc_hwcf {dilations = dense<1> : tensor<2xi64>,
H A Dtest-conv-3d-ndhwc-dhwcf-call.mlir22 linalg.conv_3d_ndhwc_dhwcf {dilations = dense<1> : tensor<3xi64>,
/llvm-project-15.0.7/mlir/docs/Dialects/Linalg/
H A DOpDSL.md167 dilations=IndexAttrDef(S.DH, S.DW, default=[1, 1])):