nvonnxparser::IOnnxConfig 类参考

配置管理器类。 更多...

#include <NvOnnxConfig.h>

公共类型
typedef int32_t	Verbosity
	定义 Verbosity 级别。 更多...

公共成员函数
virtual	~IOnnxConfig () noexcept=default
virtual void	setModelDtype (const nvinfer1::DataType) noexcept=0
	设置模型数据类型。 更多...
virtual nvinfer1::DataType	getModelDtype () const noexcept=0
	获取模型数据类型。 更多...
virtual char const *	getModelFileName () const noexcept=0
	获取模型文件名。 更多...
virtual void	setModelFileName (char const *onnxFilename) noexcept=0
	设置模型文件名。 更多...
virtual Verbosity	getVerbosityLevel () const noexcept=0
	获取 Verbosity 级别。 更多...
virtual void	addVerbosity () noexcept=0
	增加 Verbosity 级别。 更多...
virtual void	reduceVerbosity () noexcept=0
	降低 Verbosity 级别。 更多...
virtual void	setVerbosityLevel (Verbosity) noexcept=0
	设置为指定的 verbosity 级别。 更多...
virtual char const *	getTextFileName () const noexcept=0
	返回网络描述的文本文件名。 更多...
virtual void	setTextFileName (char const *textFileName) noexcept=0
	设置网络描述的文本文件名。 更多...
virtual char const *	getFullTextFileName () const noexcept=0
	获取网络描述的文本文件名，包括权重。 更多...
virtual void	setFullTextFileName (char const *fullTextFileName) noexcept=0
	设置网络描述的文本文件名，包括权重。 更多...
virtual bool	getPrintLayerInfo () const noexcept=0
	获取是否打印层信息。 更多...
virtual void	setPrintLayerInfo (bool) noexcept=0
	设置是否打印层信息。 更多...

详细描述

配置管理器类。

成员类型定义文档

Verbosity

nvonnxparser::IOnnxConfig::Verbosity

定义 Verbosity 级别。

构造函数 & 析构函数文档

~IOnnxConfig()

virtual nvonnxparser::IOnnxConfig::~IOnnxConfig ( )

virtualdefaultnoexcept

成员函数文档

addVerbosity()

virtual void nvonnxparser::IOnnxConfig::addVerbosity ( )

pure virtualnoexcept

增加 Verbosity 级别。

返回值

Verbosity 级别。

参见

reduceVerbosity(), setVerbosity(Verbosity)

getFullTextFileName()

virtual char const * nvonnxparser::IOnnxConfig::getFullTextFileName ( ) const

pure virtualnoexcept

获取网络描述的文本文件名，包括权重。

返回值

返回包含转换为纯文本的网络描述的文件名，用于调试目的。

参见

setFullTextFilename()

getModelDtype()

virtual nvinfer1::DataType nvonnxparser::IOnnxConfig::getModelDtype ( ) const

pure virtualnoexcept

获取模型数据类型。

返回值

模型的数据类型。

参见

setModelDtype() 和 DataType

getModelFileName()

virtual char const * nvonnxparser::IOnnxConfig::getModelFileName ( ) const

pure virtualnoexcept

获取模型文件名。

返回值

返回模型文件名，作为空字符结尾的 C 风格字符串。

参见

setModelFileName()

getPrintLayerInfo()

virtual bool nvonnxparser::IOnnxConfig::getPrintLayerInfo ( ) const

pure virtualnoexcept

获取是否打印层信息。

返回值

返回是否打印层信息。

参见

setPrintLayerInfo()

getTextFileName()

virtual char const * nvonnxparser::IOnnxConfig::getTextFileName ( ) const

pure virtualnoexcept

返回网络描述的文本文件名。

返回值

返回包含转换为纯文本的网络描述的文件名，用于调试目的。

参见

setTextFilename()

getVerbosityLevel()

virtual Verbosity nvonnxparser::IOnnxConfig::getVerbosityLevel ( ) const

pure virtualnoexcept

获取 Verbosity 级别。

返回值

Verbosity 级别。

参见

addVerbosity(), reduceVerbosity()

reduceVerbosity()

virtual void nvonnxparser::IOnnxConfig::reduceVerbosity ( )

pure virtualnoexcept

降低 Verbosity 级别。

参见

addVerbosity(), setVerbosity(Verbosity)

setFullTextFileName()

virtual void nvonnxparser::IOnnxConfig::setFullTextFileName ( char const *  fullTextFileName )

pure virtualnoexcept

设置网络描述的文本文件名，包括权重。

此 API 允许设置纯文本格式的网络描述文件名，等同于 ONNX protobuf。

此方法复制名称字符串。

参数

fullTextFileName

文件名。

参见

getFullTextFilename()

setModelDtype()

virtual void nvonnxparser::IOnnxConfig::setModelDtype ( const nvinfer1::DataType  )

pure virtualnoexcept

设置模型数据类型。

设置模型数据类型，以下类型之一：float -d 32 (默认), 半精度 -d 16, 和 int8 -d 8 数据类型。

参见

getModelDtype()

setModelFileName()

virtual void nvonnxparser::IOnnxConfig::setModelFileName ( char const *  onnxFilename )

pure virtualnoexcept

设置模型文件名。

模型文件名包含 ONNX pb 格式的网络描述。

此方法复制名称字符串。

参数

onnxFilename

名称。

参见

getModelFileName()

setPrintLayerInfo()

virtual void nvonnxparser::IOnnxConfig::setPrintLayerInfo ( bool  )

pure virtualnoexcept

设置是否打印层信息。

参见

getPrintLayerInfo()

setTextFileName()

virtual void nvonnxparser::IOnnxConfig::setTextFileName ( char const *  textFileName )

pure virtualnoexcept

设置网络描述的文本文件名。

此 API 允许设置纯文本格式的网络描述文件名，等同于 ONNX protobuf。

此方法复制名称字符串。

参数

textFileName

文件名。

参见

getTextFilename()

setVerbosityLevel()

virtual void nvonnxparser::IOnnxConfig::setVerbosityLevel ( Verbosity  )

pure virtualnoexcept

设置为指定的 verbosity 级别。

参见

addVerbosity(), reduceVerbosity()

---

此类文档从以下文件生成

• NvOnnxConfig.h