From e7d2ceb7bb993f6f242059e66762800f1414c64f Mon Sep 17 00:00:00 2001 From: hlu1 <14827759+hlu1@users.noreply.github.com> Date: Fri, 12 Apr 2019 18:33:45 -0700 Subject: [PATCH] [Bugfix] Fix caffe2 nnvm frontend (#2996) --- nnvm/python/nnvm/frontend/caffe2.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/nnvm/python/nnvm/frontend/caffe2.py b/nnvm/python/nnvm/frontend/caffe2.py index 367834956a10..2b3ff5a27e01 100644 --- a/nnvm/python/nnvm/frontend/caffe2.py +++ b/nnvm/python/nnvm/frontend/caffe2.py @@ -19,7 +19,7 @@ from __future__ import absolute_import as _abs import tvm from nnvm import symbol as _sym -from .common import get_nnvm_op +from .common import get_nnvm_op, Renamer, AttrConverter as AttrCvt from .onnx_caffe2_utils import dimension_picker, dimension_constraint, infer_channels, revert_caffe2_pad from . import onnx