diff --git a/python/tvm/relay/frontend/pytorch.py b/python/tvm/relay/frontend/pytorch.py index da8c0944799a..fbefd78ff2ee 100644 --- a/python/tvm/relay/frontend/pytorch.py +++ b/python/tvm/relay/frontend/pytorch.py @@ -1806,6 +1806,7 @@ def _get_convert_map(prelude): "aten::view" : _view(), "aten::reshape" : _reshape(), "aten::clone" : _clone(), + "aten::copy_" : _clone(), "aten::log_softmax" : _log_softmax(), "aten::sigmoid" : _sigmoid(), "aten::softplus" : _softplus(),