add in-place methods used by Tacotron2 to pytorch frontend (apache#8692)

Co-authored-by: Valery Chernov <[email protected]>
ylc · Sep 29, 2021 · 20a246e · 20a246e
1 parent 1e5982b
commit 20a246e
Showing 1 changed file with 2 additions and 0 deletions.
diff --git a/python/tvm/relay/frontend/pytorch.py b/python/tvm/relay/frontend/pytorch.py
@@ -2669,6 +2669,7 @@ def create_convert_map(self):
             "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 @@ def create_convert_map(self):
             "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"),