Class used for parsing models described using the UFF format. More...
Public Member Functions | |
virtual bool | registerInput (const char *inputName, nvinfer1::Dims inputDims, UffInputOrder inputOrder)=0 |
Register an input name of a UFF network with the associated Dimensions. More... | |
virtual bool | registerOutput (const char *outputName)=0 |
Register an output name of a UFF network. More... | |
virtual bool | parse (const char *file, nvinfer1::INetworkDefinition &network, nvinfer1::DataType weightsType=nvinfer1::DataType::kFLOAT)=0 |
Parse a UFF file. More... | |
virtual bool | parseBuffer (const char *buffer, std::size_t size, nvinfer1::INetworkDefinition &network, nvinfer1::DataType weightsType=nvinfer1::DataType::kFLOAT)=0 |
Parse a UFF buffer, useful if the file already live in memory. More... | |
virtual void | destroy ()=0 |
virtual int32_t | getUffRequiredVersionMajor ()=0 |
Return Version Major of the UFF. More... | |
virtual int32_t | getUffRequiredVersionMinor ()=0 |
Return Version Minor of the UFF. More... | |
virtual int32_t | getUffRequiredVersionPatch ()=0 |
Return Patch Version of the UFF. More... | |
virtual void | setPluginFactory (IPluginFactory *factory)=0 |
Set the IPluginFactory used to create the user defined plugins. More... | |
virtual void | setPluginFactoryExt (IPluginFactoryExt *factory)=0 |
Set the IPluginFactoryExt used to create the user defined pluginExts. More... | |
virtual void | setPluginNamespace (const char *libNamespace)=0 |
Set the namespace used to lookup and create plugins in the network. More... | |
virtual void | setErrorRecorder (nvinfer1::IErrorRecorder *recorder)=0 |
Set the ErrorRecorder for this interface. More... | |
virtual nvinfer1::IErrorRecorder * | getErrorRecorder () const =0 |
get the ErrorRecorder assigned to this interface. More... | |
Protected Member Functions | |
virtual | ~IUffParser () |
Class used for parsing models described using the UFF format.
|
inlineprotectedvirtual |
|
pure virtual |
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 virtual |
Register an output name of a UFF network.
outputName | Output name. |
|
pure virtual |
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 virtual |
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 virtual |
|
pure virtual |
Return Version Major of the UFF.
|
pure virtual |
Return Version Minor of the UFF.
|
pure virtual |
Return Patch Version of the UFF.
|
pure virtual |
Set the IPluginFactory used to create the user defined plugins.
factory | Pointer to an instance of the user implmentation of IPluginFactory. |
|
pure virtual |
Set the IPluginFactoryExt used to create the user defined pluginExts.
factory | Pointer to an instance of the user implmentation of IPluginFactoryExt. |
|
pure virtual |
Set the namespace used to lookup and create plugins in the network.
|
pure virtual |
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.
recorder | The error recorder to register with this interface. |
|
pure virtual |
get the ErrorRecorder assigned to this interface.
Retrieves the assigned error recorder object for the given class. A default error recorder does not exist, so a nullptr will be returned if setErrorRecorder has not been called.