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

symbol	symbol that already called subgraph api
argParams	original arg params, params needed by tensorrt will be removed after calling this function
auxParams	original aux params, params needed by tensorrt will be removed after calling this function