[Bugfix] Fix caffe2 nnvm frontend (#2996)

This commit is contained in:
hlu1 2019-04-12 18:33:45 -07:00 коммит произвёл Tianqi Chen
Родитель 8b71a28289
Коммит fefbb0061e
1 изменённых файлов: 1 добавлений и 1 удалений

Просмотреть файл

@ -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