NvOnnxParser.h 文件参考

#include "NvInfer.h"
#include <stddef.h>
#include <string>
#include <vector>

转到此文件的源代码。

类
class	nvonnxparser::IParserError
	一个包含错误信息的对象 更多...
class	nvonnxparser::IParser
	一个用于将 ONNX 模型解析为 TensorRT 网络定义的对象 更多...
class	nvonnxparser::IParserRefitter
	一个旨在重新拟合 ONNX 模型权重的接口。 更多...

命名空间
namespace	nvonnxparser
	TensorRT ONNX 解析器 API 命名空间。
namespace	nvonnxparser::anonymous_namespace{NvOnnxParser.h}

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

类型定义
typedef std::pair< std::vector< size_t >, bool >	SubGraph_t
	包含 ONNX 图中一组节点的解析能力的数据结构。 更多...
typedef std::vector< SubGraph_t >	SubGraphCollection_t
	包含从 ONNX 图中划分出的所有 SubGraph_t 的数据结构。 更多...
using	nvonnxparser::OnnxParserFlags = uint32_t
	使用二进制 OR 运算表示一个或多个 OnnxParserFlag 值，例如，1U << OnnxParserFlag::kNATIVE_INSTANCENORM。 更多...

枚举
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 }
	解析器或重拟合器可能返回的错误类型。 更多...
enum class	nvonnxparser::OnnxParserFlag : int32_t { nvonnxparser::kNATIVE_INSTANCENORM = 0 }

函数
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
	创建一个新的解析器对象。 更多...
IParserRefitter *	nvonnxparser::anonymous_namespace{NvOnnxParser.h}::createParserRefitter (nvinfer1::IRefitter &refitter, nvinfer1::ILogger &logger) noexcept
	创建一个新的 ONNX 重拟合器对象。 更多...

详细描述

这是 ONNX 解析器的 API

宏定义文档

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

类型定义文档

SubGraph_t

SubGraph_t

包含 ONNX 图中一组节点的解析能力的数据结构。

SubGraphCollection_t

SubGraphCollection_t

包含从 ONNX 图中划分出的所有 SubGraph_t 的数据结构。

函数文档

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