[Bugfix] Fix caffe2 nnvm frontend (apache#2996)

neo-ai · May 13, 2019 · e7d2ceb · e7d2ceb
1 parent 0c8b3e3
commit e7d2ceb
Showing 1 changed file with 1 addition and 1 deletion.
diff --git 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