We read every piece of feedback, and take your input very seriously.
To see all available qualifiers, see our documentation.
There was an error while loading. Please reload this page.
1 parent 00c9126 commit 5de0e39Copy full SHA for 5de0e39
python/tvm/relax/frontend/torch/fx_translator.py
@@ -1468,6 +1468,7 @@ def create_convert_map(self):
1468
"getitem": self._getitem,
1469
"contiguous": lambda node: self.env[node.args[0]],
1470
"to": self._to,
1471
+ "max_pool2d": self._max_pool2d,
1472
"avg_pool2d": self._avg_pool2d,
1473
"adaptive_avg_pool2d": self._adaptive_avg_pool2d(is_module=False),
1474
"layer_norm": self._layer_norm,
0 commit comments