This is an automated email from the ASF dual-hosted git repository.
masahi pushed a commit to branch main
in repository https://gitbox.apache.org/repos/asf/tvm.git
The following commit(s) were added to refs/heads/main by this push:
new 1abd248 add in-place methods used by Tacotron2 to pytorch frontend
(#8692)
1abd248 is described below
commit 1abd248e4a5e4d3990898fe92c789c6483a78d19
Author: Valery Chernov <[email protected]>
AuthorDate: Wed Aug 11 01:25:44 2021 +0300
add in-place methods used by Tacotron2 to pytorch frontend (#8692)
Co-authored-by: Valery Chernov <[email protected]>
---
python/tvm/relay/frontend/pytorch.py | 2 ++
1 file changed, 2 insertions(+)
diff --git a/python/tvm/relay/frontend/pytorch.py
b/python/tvm/relay/frontend/pytorch.py
index ca24335..83ee1d3 100644
--- a/python/tvm/relay/frontend/pytorch.py
+++ b/python/tvm/relay/frontend/pytorch.py
@@ -2669,6 +2669,7 @@ class PyTorchOpConverter:
"aten::clone": self.clone,
"aten::log_softmax": self.log_softmax,
"aten::sigmoid": self.sigmoid,
+ "aten::sigmoid_": self.sigmoid,
"aten::softplus": self.softplus,
"aten::avg_pool1d": self.make_avg_pool(1),
"aten::avg_pool2d": self.make_avg_pool(2),
@@ -2710,6 +2711,7 @@ class PyTorchOpConverter:
"aten::sinh": self.make_unary("sinh"),
"aten::tan": self.make_unary("tan"),
"aten::tanh": self.make_unary("tanh"),
+ "aten::tanh_": self.make_unary("tanh"),
"aten::acos": self.make_unary("acos"),
"aten::asin": self.make_unary("asin"),
"aten::atan": self.make_unary("atan"),
