|
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. |