TensorRT 8.6.1
|
Class used for parsing models described using the UFF format. More...
#include <NvUffParser.h>
Public Member Functions | |
virtual bool | registerInput (char const *inputName, nvinfer1::Dims inputDims, UffInputOrder inputOrder) noexcept=0 |
Register an input name of a UFF network with the associated Dimensions. More... | |
virtual bool | registerOutput (char const *outputName) noexcept=0 |
Register an output name of a UFF network. More... | |
virtual bool | parse (char const *file, nvinfer1::INetworkDefinition &network, nvinfer1::DataType weightsType=nvinfer1::DataType::kFLOAT) noexcept=0 |
Parse a UFF file. More... | |
virtual bool | parseBuffer (char const *buffer, std::size_t size, nvinfer1::INetworkDefinition &network, nvinfer1::DataType weightsType=nvinfer1::DataType::kFLOAT) noexcept=0 |
Parse a UFF buffer, useful if the file already live in memory. More... | |
virtual TRT_DEPRECATED void | destroy () noexcept=0 |
virtual int32_t | getUffRequiredVersionMajor () noexcept=0 |
Return Version Major of the UFF. More... | |
virtual int32_t | getUffRequiredVersionMinor () noexcept=0 |
Return Version Minor of the UFF. More... | |
virtual int32_t | getUffRequiredVersionPatch () noexcept=0 |
Return Patch Version of the UFF. More... | |
virtual void | setPluginNamespace (char const *libNamespace) noexcept=0 |
Set the namespace used to lookup and create plugins in the network. More... | |
virtual | ~IUffParser () noexcept=default |
virtual void | setErrorRecorder (nvinfer1::IErrorRecorder *recorder) noexcept=0 |
Set the ErrorRecorder for this interface. More... | |
virtual nvinfer1::IErrorRecorder * | getErrorRecorder () const noexcept=0 |
get the ErrorRecorder assigned to this interface. More... | |
Class used for parsing models described using the UFF format.
|
virtualdefaultnoexcept |
|
pure virtualnoexcept |
delete
instead. Deprecated in TRT 8.0.
|
pure virtualnoexcept |
get the ErrorRecorder assigned to this interface.
Retrieves the assigned error recorder object for the given class. A nullptr will be returned if setErrorRecorder has not been called.
|
pure virtualnoexcept |
Return Version Major of the UFF.
|
pure virtualnoexcept |
Return Version Minor of the UFF.
|
pure virtualnoexcept |
Return Patch Version of the UFF.
|
pure virtualnoexcept |
Parse a UFF file.
file | File name of the UFF file. |
network | Network in which the UFFParser will fill the layers. |
weightsType | The type on which the weights will transformed in. |
|
pure virtualnoexcept |
Parse a UFF buffer, useful if the file already live in memory.
buffer | Buffer of the UFF file. |
size | Size of buffer of the UFF file. |
network | Network in which the UFFParser will fill the layers. |
weightsType | The type on which the weights will transformed in. |
|
pure virtualnoexcept |
Register an input name of a UFF network with the associated Dimensions.
inputName | Input name. |
inputDims | Input dimensions. |
inputOrder | Input order on which the framework input was originally. |
|
pure virtualnoexcept |
Register an output name of a UFF network.
outputName | Output name. |
|
pure virtualnoexcept |
Set the ErrorRecorder for this interface.
Assigns the ErrorRecorder to this interface. The ErrorRecorder will track all errors during execution. This function will call incRefCount of the registered ErrorRecorder at least once. Setting recorder to nullptr unregisters the recorder with the interface, resulting in a call to decRefCount if a recorder has been registered.
If an error recorder is not set, messages will be sent to the global log stream.
recorder | The error recorder to register with this interface. |
|
pure virtualnoexcept |
Set the namespace used to lookup and create plugins in the network.