From 532c71935fb8e2f3d7b6424306f3c57970670708 Mon Sep 17 00:00:00 2001 From: Valery Chernov Date: Mon, 9 Aug 2021 22:32:09 +0300 Subject: [PATCH] add in-place methods used by Tacotron2 to pytorch frontend --- 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 da81280d76a3d..85b70b21d8588 100644 --- 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"),