TensorRT  6.0.1.5
nvonnxparser::IOnnxConfig Class Referenceabstract

Configuration Manager Class. More...

#include <NvOnnxConfig.h>

Public Types

typedef int Verbosity
 Defines Verbosity level.
 

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.
 
virtual void setVerbosityLevel (Verbosity)=0
 Set to specific verbosity Level.
 
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.
 

Detailed Description

Configuration Manager Class.

Member Function Documentation

◆ 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.

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

◆ getModelDtype()

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

Get the Model Data Type.

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

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

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

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

◆ getVerbosityLevel()

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

Get the Verbosity Level.

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

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

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

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

◆ setPrintLayerInfo()

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

Set whether the layer information will be printed.

See also
getPrintLayerInfo()

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

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