TensorRT
|
Class used for parsing models described using the UFF format. More...
#include <NvUffParser.h>
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 int | getUffRequiredVersionMajor ()=0 |
Return Version Major of the UFF. | |
virtual int | getUffRequiredVersionMinor ()=0 |
Return Version Minor of the UFF. | |
virtual int | getUffRequiredVersionPatch ()=0 |
Return Patch Version of the UFF. | |
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... | |
Class used for parsing models described using the UFF format.
|
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 |
Register an input name of a UFF network with the associated Dimensions.
inputName | Input name. |
inputDims | Input Dimensions, always provide your dimensions in CHW even if your network input was in HWC in yout original framework. |
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 |
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. |