You are viewing a plain text version of this content. The canonical link for it is here.
Posted to commits@mxnet.apache.org by ma...@apache.org on 2018/08/10 12:19:37 UTC
[incubator-mxnet] branch master updated: [MXNET-703] Use relative
path for symbol import (#12124)
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 <ke...@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 '