ó ùµÈ[c@s—dZddlZddlZddlZddlmZddlmZm Z m Z ddlm Z d„Z d„Z d „Zdddd „ZdS( s7 Module to enable the use of TensorRT optimized graphs.iÿÿÿÿNi(tsymbol(t_LIBt SymbolHandlet MXNetError(t check_callcCstt|ƒƒtjdndarray A dictionary of mappings from parameter names to parameter NDArrays. type_dict : Dict of str->numpy.dtype Input type dictionary, name->dtype stype_dict : Dict of str->str Input storage type dictionary, name->storage_type group2ctx : Dict of string to mx.Context The dict mapping the `ctx_group` attribute to the context assignment. kwargs : Dict of str->shape Input shape dictionary, name->shape Returns ------- executor : mxnet.Executor An optimized TensorRT executor. t shared_buffert type_dictt stype_dictt group2ctx(t simple_bind(Rtctxt all_paramsRRRtkwargs((sV/usr/local/lib/python2.7/site-packages/mxnet-1.3.1-py2.7.egg/mxnet/contrib/tensorrt.pyt tensorrt_bindLs (t__doc__RRRtRRtbaseRRRRR RRtNoneR"(((sV/usr/local/lib/python2.7/site-packages/mxnet-1.3.1-py2.7.egg/mxnet/contrib/tensorrt.pyts