TensorRT 8.6.1
NvOnnxParser.h File Reference
#include "NvInfer.h"
#include <stddef.h>
#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...
 

Namespaces

namespace  nvonnxparser
 The TensorRT ONNX parser API namespace.
 

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
}
 The type of error that the parser 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 int getNvOnnxParserVersion ()
 

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 
)

◆ getNvOnnxParserVersion()

TENSORRTAPI int getNvOnnxParserVersion ( )

  Copyright © 2024 NVIDIA Corporation
  Privacy Policy | Manage My Privacy | Do Not Sell or Share My Data | Terms of Service | Accessibility | Corporate Policies | Product Security | Contact