nvonnxparser 命名空间参考

TensorRT ONNX 解析器 API 命名空间。 更多...

命名空间
namespace	anonymous_namespace{NvOnnxParser.h}

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

类型定义
using	OnnxParserFlags = uint32_t
	使用二进制 OR 运算表示一个或多个 OnnxParserFlag 值，例如，1U << OnnxParserFlag::kNATIVE_INSTANCENORM。 更多...

枚举
enum class	ErrorCode : int {   kSUCCESS = 0 , kINTERNAL_ERROR = 1 , kMEM_ALLOC_FAILED = 2 , kMODEL_DESERIALIZE_FAILED = 3 ,   kINVALID_VALUE = 4 , kINVALID_GRAPH = 5 , kINVALID_NODE = 6 , kUNSUPPORTED_GRAPH = 7 ,   kUNSUPPORTED_NODE = 8 , kUNSUPPORTED_NODE_ATTR = 9 , kUNSUPPORTED_NODE_INPUT = 10 , kUNSUPPORTED_NODE_DATATYPE = 11 ,   kUNSUPPORTED_NODE_DYNAMIC = 12 , kUNSUPPORTED_NODE_SHAPE = 13 , kREFIT_FAILED = 14 }
	解析器或重拟合器可能返回的错误类型。 更多...
enum class	OnnxParserFlag : int32_t { kNATIVE_INSTANCENORM = 0 }

函数
IOnnxConfig *	createONNXConfig ()
template<typename T >
constexpr int32_t	EnumMax () noexcept
template<>
constexpr int32_t	EnumMax< ErrorCode > () noexcept
template<>
constexpr int32_t	EnumMax< OnnxParserFlag > () noexcept

详细描述

TensorRT ONNX 解析器 API 命名空间。

类型定义文档

OnnxParserFlags

using nvonnxparser::OnnxParserFlags = typedef uint32_t

使用二进制 OR 运算表示一个或多个 OnnxParserFlag 值，例如，1U << OnnxParserFlag::kNATIVE_INSTANCENORM。

另请参阅

IParser::setFlags() 和 IParser::getFlags()

枚举类型文档

ErrorCode

enum class nvonnxparser::ErrorCode : int

strong

解析器或重拟合器可能返回的错误类型。

枚举器
kSUCCESS
kINTERNAL_ERROR
kMEM_ALLOC_FAILED
kMODEL_DESERIALIZE_FAILED
kINVALID_VALUE
kINVALID_GRAPH
kINVALID_NODE
kUNSUPPORTED_GRAPH
kUNSUPPORTED_NODE
kUNSUPPORTED_NODE_ATTR
kUNSUPPORTED_NODE_INPUT
kUNSUPPORTED_NODE_DATATYPE
kUNSUPPORTED_NODE_DYNAMIC
kUNSUPPORTED_NODE_SHAPE
kREFIT_FAILED

OnnxParserFlag

enum class nvonnxparser::OnnxParserFlag : int32_t

strong

枚举器
kNATIVE_INSTANCENORM	将 ONNX 模型解析到 INetworkDefinition 中，目的是为 InstanceNormalization 节点使用 TensorRT 的原生层实现，而不是插件实现。构建版本兼容或硬件兼容引擎时，需要此标志。默认情况下，此标志设置为开启。

函数文档

createONNXConfig()

IOnnxConfig * nvonnxparser::createONNXConfig

(

)

EnumMax()

template<typename T >

constexpr int32_t nvonnxparser::EnumMax ( )

inlineconstexprnoexcept

EnumMax< ErrorCode >()

template<>

constexpr int32_t nvonnxparser::EnumMax< ErrorCode > ( )

inlineconstexprnoexcept

ErrorCode 枚举中标志的最大数量。

另请参阅

ErrorCode

EnumMax< OnnxParserFlag >()

template<>

constexpr int32_t nvonnxparser::EnumMax< OnnxParserFlag > ( )

inlineconstexprnoexcept

OnnxParserFlag 枚举中标志的最大数量。

另请参阅

OnnxParserFlag