add aten::masked_fill_ in pytorch frontend (#8403)

Co-authored-by: Jackson Hsieh <chengpi@amazon.com>
apache · Jul 5, 2021 · ec47129 · ec47129
1 parent 6a3d950
commit ec47129
Showing 1 changed file with 1 addition and 0 deletions.
diff --git a/python/tvm/relay/frontend/pytorch.py b/python/tvm/relay/frontend/pytorch.py
@@ -2537,6 +2537,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,