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

Co-authored-by: Jackson Hsieh <[email protected]>
ylc · Sep 29, 2021 · 239193b · 239193b
1 parent 5625f84
commit 239193b
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
@@ -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,