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 | List of all members
nvonnxparser::IOnnxConfig Class Referenceabstract

Configuration Manager Class. More...

Inheritance diagram for nvonnxparser::IOnnxConfig:
Inheritance graph
[legend]

Public Types

typedef int Verbosity
 Defines Verbosity level. More...
 

Public Member Functions

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

Protected Member Functions

virtual ~IOnnxConfig ()
 

Detailed Description

Configuration Manager Class.

Member Typedef Documentation

◆ Verbosity

nvonnxparser::IOnnxConfig::Verbosity

Defines Verbosity level.

Constructor & Destructor Documentation

◆ ~IOnnxConfig()

virtual nvonnxparser::IOnnxConfig::~IOnnxConfig ( )
inlineprotectedvirtual

Member Function Documentation

◆ setModelDtype()

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

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()

Implemented in ParserOnnxConfig, and SampleConfig.

◆ getModelDtype()

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

Get the Model Data Type.

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

Implemented in SampleConfig, and ParserOnnxConfig.

◆ getModelFileName()

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

Get the Model FileName.

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

Implemented in SampleConfig, and ParserOnnxConfig.

◆ setModelFileName()

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

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()

Implemented in SampleConfig, and ParserOnnxConfig.

◆ getVerbosityLevel()

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

Get the Verbosity Level.

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

Implemented in SampleConfig, and ParserOnnxConfig.

◆ addVerbosity()

virtual void nvonnxparser::IOnnxConfig::addVerbosity ( )
pure virtual

Increase the Verbosity Level.

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

Implemented in SampleConfig, and ParserOnnxConfig.

◆ reduceVerbosity()

virtual void nvonnxparser::IOnnxConfig::reduceVerbosity ( )
pure virtual

Decrease verbosity Level.

Implemented in SampleConfig, and ParserOnnxConfig.

◆ setVerbosityLevel()

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

Set to specific verbosity Level.

Implemented in ParserOnnxConfig.

◆ getTextFileName()

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

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()

Implemented in SampleConfig, and ParserOnnxConfig.

◆ setTextFileName()

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

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()

Implemented in SampleConfig, and ParserOnnxConfig.

◆ getFullTextFileName()

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

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()

Implemented in SampleConfig, and ParserOnnxConfig.

◆ setFullTextFileName()

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

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()

Implemented in SampleConfig, and ParserOnnxConfig.

◆ getPrintLayerInfo()

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

Get whether the layer information will be printed.

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

Implemented in SampleConfig, and ParserOnnxConfig.

◆ setPrintLayerInfo()

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

Set whether the layer information will be printed.

See also
getPrintLayerInfo()

◆ destroy()

virtual void nvonnxparser::IOnnxConfig::destroy ( )
pure virtual

Destroy IOnnxConfig object.

Implemented in SampleConfig, and ParserOnnxConfig.

The documentation for this class was generated from the following file: