TensorRT
 All Classes Namespaces Files Functions Variables Typedefs Enumerations Enumerator Macros Pages
nvuffparser::IUffParser Class Referenceabstract

Public Member Functions

virtual bool registerInput (const char *inputName, nvinfer1::DimsCHW inputDims, UffInputOrder inputOrder=UffInputOrder::kNCHW)=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.
 

Member Function Documentation

virtual bool nvuffparser::IUffParser::parse ( const char *  file,
nvinfer1::INetworkDefinition network,
nvinfer1::DataType  weightsType = nvinfer1::DataType::kFLOAT 
)
pure virtual

parse a UFF file

Parameters
filefile name of the UFF file
networknetwork in which the UFFParser will fill the layers
weightsTypethe type on which the weights will transformed in
virtual bool nvuffparser::IUffParser::parseBuffer ( const char *  buffer,
std::size_t  size,
nvinfer1::INetworkDefinition network,
nvinfer1::DataType  weightsType = nvinfer1::DataType::kFLOAT 
)
pure virtual

parse a UFF buffer, useful if the file already live in memory

Parameters
bufferbuffer of the UFF file
sizesize of buffer of the UFF file
networknetwork in which the UFFParser will fill the layers
weightsTypethe type on which the weights will transformed in
virtual bool nvuffparser::IUffParser::registerInput ( const char *  inputName,
nvinfer1::DimsCHW  inputDims,
UffInputOrder  inputOrder = UffInputOrder::kNCHW 
)
pure virtual

register an input name of a UFF network with the associated Dimensions

Parameters
inputNameinput name
inputDimsinput Dimensions, always provide your dimensions in CHW even if your network input was in HWC in yout original framework.
inputOrderinput order on which the framework input was originally
virtual bool nvuffparser::IUffParser::registerOutput ( const char *  outputName)
pure virtual

register an output name of a UFF network

Parameters
outputNameoutput name

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