This is an automated email from the ASF dual-hosted git repository.

marcoabreu pushed a commit to branch master
in repository https://gitbox.apache.org/repos/asf/incubator-mxnet.git


The following commit(s) were added to refs/heads/master by this push:
     new 53703e2  [MXNET-703] Use relative path for symbol import (#12124)
53703e2 is described below

commit 53703e2d4bd0ea1b87472dfcf00890a17ee6ce39
Author: Kellen Sunderland <kellen.sunderl...@gmail.com>
AuthorDate: Fri Aug 10 14:19:28 2018 +0200

    [MXNET-703] Use relative path for symbol import (#12124)
---
 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 bb20767..4ff39c4 100644
--- a/python/mxnet/contrib/tensorrt.py
+++ b/python/mxnet/contrib/tensorrt.py
@@ -21,7 +21,7 @@ import ctypes
 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 '

Reply via email to