From 11aa1277554248aaf093682016411aeebf2bf510 Mon Sep 17 00:00:00 2001 From: Kellen Sunderland Date: Fri, 10 Aug 2018 09:54:16 +0200 Subject: [PATCH] [MXNET-703] Use relative path for symbol import --- python/mxnet/contrib/tensorrt.py | 4 ++-- 1 file changed, 2 insertions(+), 2 deletions(-) diff --git a/python/mxnet/contrib/tensorrt.py b/python/mxnet/contrib/tensorrt.py index bb20767b3e6f..4ff39c4b4829 100644 --- a/python/mxnet/contrib/tensorrt.py +++ b/python/mxnet/contrib/tensorrt.py @@ -21,7 +21,7 @@ import logging import os -from mxnet.symbol import Symbol +from .. import symbol as sym from ..base import _LIB, SymbolHandle, MXNetError from ..base import check_call @@ -65,7 +65,7 @@ def get_optimized_symbol(executor): handle = SymbolHandle() try: check_call(_LIB.MXExecutorGetOptimizedSymbol(executor.handle, ctypes.byref(handle))) - result = Symbol(handle=handle) + result = sym.Symbol(handle=handle) return result except MXNetError: logging.error('Error while trying to fetch TRT optimized symbol for graph. Please ensure '