TensorRT
|
ONNX Parser Class. More...
#include <NvOnnxParser.h>
Public Member Functions | |
virtual StatusCode | parse ()=0 |
Parse the ONNX Model. | |
virtual StatusCode | parse (const char *onnx_filename, nvinfer1::DataType dataType)=0 |
Parse the ONNX Model. More... | |
virtual StatusCode | reportParsingInfo ()=0 |
Report parsing info. | |
virtual StatusCode | convertToTRTNetwork ()=0 |
Convert ONNX model to TensorRT network. | |
virtual StatusCode | writeText () const =0 |
Write ASCII Equivalent of ONNX model protbuf file. | |
virtual StatusCode | writeTextWithWeights () const =0 |
Write ASCII Equivalent of ONNX model protbuf file includeing the weights. | |
virtual nvinfer1::INetworkDefinition * | getTRTNetwork () const =0 |
Returns the TRT Network. More... | |
virtual nvinfer1::ILogger * | getLogger ()=0 |
Returns the Logger. More... | |
virtual void | destroy ()=0 |
ONNX Parser Class.
|
pure virtual |
Returns the Logger.
|
pure virtual |
Returns the TRT Network.
|
pure virtual |
Parse the ONNX Model.
onnx_filename | Onnx Model Filename. |
dataType | The data type. |