From 76d16de0ab5a201577fcb31e142ea197c9bc3596 Mon Sep 17 00:00:00 2001 From: huajsj Date: Mon, 8 Jun 2020 18:17:57 -0700 Subject: [PATCH] [Pytorch] add support for operator copy_ --- python/tvm/relay/frontend/pytorch.py | 1 + 1 file changed, 1 insertion(+) 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(),