diff options
author | natashaknk <natashaknk@google.com> | 2021-08-12 15:37:34 -0700 |
---|---|---|
committer | Rob Suderman <rob.suderman@gmail.com> | 2021-08-12 15:43:41 -0700 |
commit | ba0997ca09d3a61a7b67e3b757136ad0d1fcac26 (patch) | |
tree | 1d8e9b66a82aa1faf5fe203aa6bf28a52e01790e /mlir/python | |
parent | 2ff7ca98a99bbfc4f44b8ba1e2e4e594f8ee253e (diff) | |
download | llvm-ba0997ca09d3a61a7b67e3b757136ad0d1fcac26.zip llvm-ba0997ca09d3a61a7b67e3b757136ad0d1fcac26.tar.gz llvm-ba0997ca09d3a61a7b67e3b757136ad0d1fcac26.tar.bz2 |
[mlir][tosa] Fix depthwise_conv2D strides/dilation and name
Reviewed By: rsuderman
Differential Revision: https://reviews.llvm.org/D107997
Diffstat (limited to 'mlir/python')
-rw-r--r-- | mlir/python/mlir/dialects/linalg/opdsl/ops/core_named_ops.py | 4 |
1 files changed, 2 insertions, 2 deletions
diff --git a/mlir/python/mlir/dialects/linalg/opdsl/ops/core_named_ops.py b/mlir/python/mlir/dialects/linalg/opdsl/ops/core_named_ops.py index b9faeeb..b2bd4ce 100644 --- a/mlir/python/mlir/dialects/linalg/opdsl/ops/core_named_ops.py +++ b/mlir/python/mlir/dialects/linalg/opdsl/ops/core_named_ops.py @@ -216,7 +216,7 @@ def conv_2d_nhwc_hwcf_q( ]) - cast(U, IZp)) * (cast(U, K[D.kh, D.kw, D.c, D.f]) - cast(U, KZp)) @linalg_structured_op -def depthwise_conv2D_nchw( +def depthwise_conv2D_nhwc( I=TensorDef(T1, S.N, S.IH, S.IW, S.IC), K=TensorDef(T2, S.KH, S.KW, S.IC, S.CM), O=TensorDef(U, S.N, S.OH, S.OW, S.IC, S.CM, output=True), @@ -233,7 +233,7 @@ def depthwise_conv2D_nchw( D.ic]) * cast(U, K[D.kh, D.kw, D.ic, D.cm]) @linalg_structured_op -def depthwise_conv2D_nchw_q( +def depthwise_conv2D_nhwc_q( I=TensorDef(T1, S.N, S.IH, S.IW, S.IC), K=TensorDef(T2, S.KH, S.KW, S.IC, S.CM), IZp=ScalarDef(I32), |