Skip to content

Commit 5de0e39

Browse files
committed
add maxpool2d to call_function
1 parent 00c9126 commit 5de0e39

File tree

1 file changed

+1
-0
lines changed

1 file changed

+1
-0
lines changed

python/tvm/relax/frontend/torch/fx_translator.py

Lines changed: 1 addition & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -1468,6 +1468,7 @@ def create_convert_map(self):
14681468
"getitem": self._getitem,
14691469
"contiguous": lambda node: self.env[node.args[0]],
14701470
"to": self._to,
1471+
"max_pool2d": self._max_pool2d,
14711472
"avg_pool2d": self._avg_pool2d,
14721473
"adaptive_avg_pool2d": self._adaptive_avg_pool2d(is_module=False),
14731474
"layer_norm": self._layer_norm,

0 commit comments

Comments
 (0)