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_t >	SubGraphCollection_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 () noexcept
template<>
constexpr int32_t	nvonnxparser::EnumMax< ErrorCode > () noexcept
template<>
constexpr int32_t	nvonnxparser::EnumMax< OnnxParserFlag > () noexcept
TENSORRTAPI void *	createNvOnnxParser_INTERNAL (void *network, void *logger, int version) noexcept
TENSORRTAPI void *	createNvOnnxParserRefitter_INTERNAL (void *refitter, void *logger, int32_t version) noexcept
TENSORRTAPI int	getNvOnnxParserVersion () noexcept
IParser *	nvonnxparser::anonymous_namespace{NvOnnxParser.h}::createParser (nvinfer1::INetworkDefinition &network, nvinfer1::ILogger &logger) noexcept
	Create a new parser object. More...
IParserRefitter *	nvonnxparser::anonymous_namespace{NvOnnxParser.h}::createParserRefitter (nvinfer1::IRefitter &refitter, nvinfer1::ILogger &logger) noexcept
	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

SubGraph_t

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

SubGraphCollection_t

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  )

noexcept

createNvOnnxParserRefitter_INTERNAL()

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

noexcept

getNvOnnxParserVersion()

TENSORRTAPI int getNvOnnxParserVersion ( )

noexcept