TensorRT  7.2.1.6
NVIDIA TensorRT
Looking for a C++ dev who knows TensorRT?
I'm looking for work. Hire me!
Public Types | Public Member Functions | Protected Member Functions | Protected Attributes | List of all members
ParserOnnxConfig Class Referenceabstract

Configuration Manager Class Concrete Implementation. More...

Inheritance diagram for ParserOnnxConfig:
Inheritance graph
[legend]
Collaboration diagram for ParserOnnxConfig:
Collaboration graph
[legend]

Public Types

typedef int Verbosity
 Defines Verbosity level. More...
 

Public Member Functions

 ParserOnnxConfig ()
 
virtual void setModelDtype (const nvinfer1::DataType modelDtype)
 Set the Model Data Type. More...
 
virtual nvinfer1::DataType getModelDtype () const
 Get the Model Data Type. More...
 
virtual const char * getModelFileName () const
 Get the Model FileName. More...
 
virtual void setModelFileName (const char *onnxFilename)
 Set the Model File Name. More...
 
virtual nvonnxparser::IOnnxConfig::Verbosity getVerbosityLevel () const
 Get the Verbosity Level. More...
 
virtual void addVerbosity ()
 Increase the Verbosity Level. More...
 
virtual void reduceVerbosity ()
 Decrease verbosity Level. More...
 
virtual void setVerbosityLevel (nvonnxparser::IOnnxConfig::Verbosity verbosity)
 Set to specific verbosity Level. More...
 
virtual const char * getTextFileName () const
 Returns the File Name of the Network Description as a Text File. More...
 
virtual void setTextFileName (const char *textFilename)
 Set the File Name of the Network Description as a Text File. More...
 
virtual const char * getFullTextFileName () const
 Get the File Name of the Network Description as a Text File, including the weights. More...
 
virtual void setFullTextFileName (const char *fullTextFilename)
 Set the File Name of the Network Description as a Text File, including the weights. More...
 
virtual bool getPrintLayerInfo () const
 Get whether the layer information will be printed. More...
 
virtual bool isDebug () const
 
virtual void destroy ()
 Destroy IOnnxConfig object. More...
 
virtual void setPrintLayerInfo (bool)=0
 Set whether the layer information will be printed. More...
 

Protected Member Functions

 ~ParserOnnxConfig ()
 

Protected Attributes

string mModelFilename {}
 
string mTextFilename {}
 
string mFullTextFilename {}
 
nvinfer1::DataType mModelDtype
 
nvonnxparser::IOnnxConfig::Verbosity mVerbosity
 
bool mPrintLayercInfo
 

Detailed Description

Configuration Manager Class Concrete Implementation.

Note
:

Member Typedef Documentation

◆ Verbosity

nvonnxparser::IOnnxConfig::Verbosity
inherited

Defines Verbosity level.

Constructor & Destructor Documentation

◆ ParserOnnxConfig()

ParserOnnxConfig::ParserOnnxConfig ( )
inline

◆ ~ParserOnnxConfig()

ParserOnnxConfig::~ParserOnnxConfig ( )
inlineprotected

Member Function Documentation

◆ setModelDtype()

virtual void ParserOnnxConfig::setModelDtype ( const nvinfer1::DataType  )
inlinevirtual

Set the Model Data Type.

Sets the Model DataType, one of the following: float -d 32 (default), half precision -d 16, and int8 -d 8 data types.

See also
getModelDtype()

Implements nvonnxparser::IOnnxConfig.

◆ getModelDtype()

virtual nvinfer1::DataType ParserOnnxConfig::getModelDtype ( ) const
inlinevirtual

Get the Model Data Type.

Returns
DataType nvinfer1::DataType
See also
setModelDtype() and #DataType

Implements nvonnxparser::IOnnxConfig.

◆ getModelFileName()

virtual const char* ParserOnnxConfig::getModelFileName ( ) const
inlinevirtual

Get the Model FileName.

Returns
Return the Model Filename, as a pointer to a NULL-terminated character sequence.
See also
setModelFileName()

Implements nvonnxparser::IOnnxConfig.

◆ setModelFileName()

virtual void ParserOnnxConfig::setModelFileName ( const char *  onnxFilename)
inlinevirtual

Set the Model File Name.

The Model File name contains the Network Description in ONNX pb format.

This method copies the name string.

Parameters
onnxFilenameThe name.
See also
getModelFileName()

Implements nvonnxparser::IOnnxConfig.

◆ getVerbosityLevel()

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

Get the Verbosity Level.

Returns
The Verbosity Level.
See also
addVerbosity(), reduceVerbosity()

Implements nvonnxparser::IOnnxConfig.

◆ addVerbosity()

virtual void ParserOnnxConfig::addVerbosity ( )
inlinevirtual

Increase the Verbosity Level.

Returns
The Verbosity Level.
See also
addVerbosity(), reduceVerbosity(), setVerbosity(Verbosity) Increase verbosity Level.

Implements nvonnxparser::IOnnxConfig.

◆ reduceVerbosity()

virtual void ParserOnnxConfig::reduceVerbosity ( )
inlinevirtual

Decrease verbosity Level.

Implements nvonnxparser::IOnnxConfig.

◆ setVerbosityLevel()

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

Set to specific verbosity Level.

Implements nvonnxparser::IOnnxConfig.

◆ getTextFileName()

virtual const char* ParserOnnxConfig::getTextFileName ( ) const
inlinevirtual

Returns the File Name of the Network Description as a Text File.

Returns
Return the name of the file containing the network description converted to a plain text, used for debugging purposes.
See also
setTextFilename()

Implements nvonnxparser::IOnnxConfig.

◆ setTextFileName()

virtual void ParserOnnxConfig::setTextFileName ( const char *  textFileName)
inlinevirtual

Set the File Name of the Network Description as a Text File.

This API allows setting a file name for the network description in plain text, equivalent of the ONNX protobuf.

This method copies the name string.

Parameters
textFileNameName of the file.
See also
getTextFilename()

Implements nvonnxparser::IOnnxConfig.

◆ getFullTextFileName()

virtual const char* ParserOnnxConfig::getFullTextFileName ( ) const
inlinevirtual

Get the File Name of the Network Description as a Text File, including the weights.

Returns
Return the name of the file containing the network description converted to a plain text, used for debugging purposes.
See also
setFullTextFilename()

Implements nvonnxparser::IOnnxConfig.

◆ setFullTextFileName()

virtual void ParserOnnxConfig::setFullTextFileName ( const char *  fullTextFileName)
inlinevirtual

Set the File Name of the Network Description as a Text File, including the weights.

This API allows setting a file name for the network description in plain text, equivalent of the ONNX protobuf.

This method copies the name string.

Parameters
fullTextFileNameName of the file.
See also
getFullTextFilename()

Implements nvonnxparser::IOnnxConfig.

◆ getPrintLayerInfo()

virtual bool ParserOnnxConfig::getPrintLayerInfo ( ) const
inlinevirtual

Get whether the layer information will be printed.

Returns
Returns whether the layer information will be printed.
See also
setPrintLayerInfo()

Implements nvonnxparser::IOnnxConfig.

◆ isDebug()

virtual bool ParserOnnxConfig::isDebug ( ) const
inlinevirtual

◆ destroy()

virtual void ParserOnnxConfig::destroy ( )
inlinevirtual

Destroy IOnnxConfig object.

Implements nvonnxparser::IOnnxConfig.

◆ setPrintLayerInfo()

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

Set whether the layer information will be printed.

See also
getPrintLayerInfo()

Member Data Documentation

◆ mModelFilename

string ParserOnnxConfig::mModelFilename {}
protected

◆ mTextFilename

string ParserOnnxConfig::mTextFilename {}
protected

◆ mFullTextFilename

string ParserOnnxConfig::mFullTextFilename {}
protected

◆ mModelDtype

nvinfer1::DataType ParserOnnxConfig::mModelDtype
protected

◆ mVerbosity

nvonnxparser::IOnnxConfig::Verbosity ParserOnnxConfig::mVerbosity
protected

◆ mPrintLayercInfo

bool ParserOnnxConfig::mPrintLayercInfo
protected
The documentation for this class was generated from the following file: