mxnet
Functions
mxnet::cpp::contrib Namespace Reference

Functions

void InitTensorRTParams (const mxnet::cpp::Symbol &symbol, std::map< std::string, mxnet::cpp::NDArray > *argParams, std::map< std::string, mxnet::cpp::NDArray > *auxParams)
 

Function Documentation

void mxnet::cpp::contrib::InitTensorRTParams ( const mxnet::cpp::Symbol symbol,
std::map< std::string, mxnet::cpp::NDArray > *  argParams,
std::map< std::string, mxnet::cpp::NDArray > *  auxParams 
)
inline

this is a mimic to https://github.com/apache/incubator-mxnet/blob/master/python/mxnet/contrib/tensorrt.py#L37

Parameters
symbolsymbol that already called subgraph api
argParamsoriginal arg params, params needed by tensorrt will be removed after calling this function
auxParamsoriginal aux params, params needed by tensorrt will be removed after calling this function