mxnet
|
Functions | |
void | 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/mxnet/blob/master/python/mxnet/contrib/tensorrt.py#L37
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 |