From b8adb109a9753d75c92bbd4d8b1e9e38b5e7a7d5 Mon Sep 17 00:00:00 2001 From: hlu1 <14827759+hlu1@users.noreply.github.com> Date: Tue, 9 Apr 2019 15:36:57 -0700 Subject: [PATCH] [Bugfix] Fix caffe2 nnvm frontend --- 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 63b7913dd755..2356e0f52002 100755 --- a/nnvm/python/nnvm/frontend/caffe2.py +++ b/nnvm/python/nnvm/frontend/caffe2.py @@ -3,7 +3,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