TensorRT 10.0.0
NvOnnxParser.h File Reference
#include "NvInfer.h"
#include <stddef.h>
#include <string>
#include <vector>

Go to the source code of this file.

Classes

class  nvonnxparser::IParserError
 an object containing information about an error More...
 
class  nvonnxparser::IParser
 an object for parsing ONNX models into a TensorRT network definition More...
 
class  nvonnxparser::IParserRefitter
 An interface designed to refit weights from an ONNX model. More...
 

Namespaces

namespace  nvonnxparser
 The TensorRT ONNX parser API namespace.
 
namespace  nvonnxparser::anonymous_namespace{NvOnnxParser.h}
 

Macros

#define NV_ONNX_PARSER_MAJOR   0
 
#define NV_ONNX_PARSER_MINOR   1
 
#define NV_ONNX_PARSER_PATCH   0
 

Typedefs

typedef std::pair< std::vector< size_t >, bool > SubGraph_t
 The data structure containing the parsing capability of a set of nodes in an ONNX graph. More...
 
typedef std::vector< SubGraph_tSubGraphCollection_t
 The data structure containing all SubGraph_t partitioned out of an ONNX graph. More...
 
using nvonnxparser::OnnxParserFlags = uint32_t
 Represents one or more OnnxParserFlag values using binary OR operations, e.g., 1U << OnnxParserFlag::kNATIVE_INSTANCENORM. More...
 

Enumerations

enum class  nvonnxparser::ErrorCode : int {
  nvonnxparser::kSUCCESS = 0 , nvonnxparser::kINTERNAL_ERROR = 1 , nvonnxparser::kMEM_ALLOC_FAILED = 2 , nvonnxparser::kMODEL_DESERIALIZE_FAILED = 3 ,
  nvonnxparser::kINVALID_VALUE = 4 , nvonnxparser::kINVALID_GRAPH = 5 , nvonnxparser::kINVALID_NODE = 6 , nvonnxparser::kUNSUPPORTED_GRAPH = 7 ,
  nvonnxparser::kUNSUPPORTED_NODE = 8 , nvonnxparser::kUNSUPPORTED_NODE_ATTR = 9 , nvonnxparser::kUNSUPPORTED_NODE_INPUT = 10 , nvonnxparser::kUNSUPPORTED_NODE_DATATYPE = 11 ,
  nvonnxparser::kUNSUPPORTED_NODE_DYNAMIC = 12 , nvonnxparser::kUNSUPPORTED_NODE_SHAPE = 13 , nvonnxparser::kREFIT_FAILED = 14
}
 The type of error that the parser or refitter may return. More...
 
enum class  nvonnxparser::OnnxParserFlag : int32_t { nvonnxparser::kNATIVE_INSTANCENORM = 0 }
 

Functions

template<typename T >
constexpr int32_t nvonnxparser::EnumMax ()
 
template<>
constexpr int32_t nvonnxparser::EnumMax< ErrorCode > ()
 
template<>
constexpr int32_t nvonnxparser::EnumMax< OnnxParserFlag > ()
 
TENSORRTAPI void * createNvOnnxParser_INTERNAL (void *network, void *logger, int version)
 
TENSORRTAPI void * createNvOnnxParserRefitter_INTERNAL (void *refitter, void *logger, int32_t version)
 
TENSORRTAPI int getNvOnnxParserVersion ()
 
IParser * nvonnxparser::anonymous_namespace{NvOnnxParser.h}::createParser (nvinfer1::INetworkDefinition &network, nvinfer1::ILogger &logger)
 Create a new parser object. More...
 
IParserRefitter * nvonnxparser::anonymous_namespace{NvOnnxParser.h}::createParserRefitter (nvinfer1::IRefitter &refitter, nvinfer1::ILogger &logger)
 Create a new ONNX refitter object. More...
 

Detailed Description

This is the API for the ONNX Parser

Macro Definition Documentation

◆ NV_ONNX_PARSER_MAJOR

#define NV_ONNX_PARSER_MAJOR   0

◆ NV_ONNX_PARSER_MINOR

#define NV_ONNX_PARSER_MINOR   1

◆ NV_ONNX_PARSER_PATCH

#define NV_ONNX_PARSER_PATCH   0

Typedef Documentation

◆ SubGraph_t

The data structure containing the parsing capability of a set of nodes in an ONNX graph.

◆ SubGraphCollection_t

The data structure containing all SubGraph_t partitioned out of an ONNX graph.

Function Documentation

◆ createNvOnnxParser_INTERNAL()

TENSORRTAPI void * createNvOnnxParser_INTERNAL ( void *  network,
void *  logger,
int  version 
)

◆ createNvOnnxParserRefitter_INTERNAL()

TENSORRTAPI void * createNvOnnxParserRefitter_INTERNAL ( void *  refitter,
void *  logger,
int32_t  version 
)

◆ getNvOnnxParserVersion()

TENSORRTAPI int getNvOnnxParserVersion ( )