From 239193b49b430d0f3f03251ee0f4facc17c0a183 Mon Sep 17 00:00:00 2001 From: hsiehjackson <37269846+hsiehjackson@users.noreply.github.com> Date: Tue, 6 Jul 2021 05:27:53 +0800 Subject: [PATCH] add aten::masked_fill_ in pytorch frontend (#8403) Co-authored-by: Jackson Hsieh --- python/tvm/relay/frontend/pytorch.py | 1 + 1 file changed, 1 insertion(+) diff --git a/python/tvm/relay/frontend/pytorch.py b/python/tvm/relay/frontend/pytorch.py index ac18e68dbddf..b04a84da254c 100644 --- a/python/tvm/relay/frontend/pytorch.py +++ b/python/tvm/relay/frontend/pytorch.py @@ -2546,6 +2546,7 @@ def create_convert_map(self): "aten::hardsigmoid": self.hard_sigmoid, "aten::cumsum": self.cumsum, "aten::masked_fill": self.masked_fill, + "aten::masked_fill_": self.masked_fill, "aten::masked_select": self.masked_select, "aten::argsort": self.argsort, "aten::sort": self.sort,