Go to the documentation of this file.
50 #ifndef NV_UFF_PARSER_H
51 #define NV_UFF_PARSER_H
62 #define UFF_REQUIRED_VERSION_MAJOR 0
63 #define UFF_REQUIRED_VERSION_MINOR 6
64 #define UFF_REQUIRED_VERSION_PATCH 9
117 FieldMap(
const char* name,
const void* data,
const FieldType type, int32_t length = 1);
257 extern "C" TENSORRTAPI
void* createNvUffParser_INTERNAL() noexcept;
Definition: NvUffParser.h:120
A network definition for input to the builder.
Definition: NvInfer.h:5450
FieldType
The possible field types for custom layer.
Definition: NvUffParser.h:90
virtual int32_t getUffRequiredVersionMinor() noexcept=0
Return Version Minor of the UFF.
An array of field params used as a layer parameter for plugin layers.
Definition: NvUffParser.h:109
virtual void setErrorRecorder(nvinfer1::IErrorRecorder *recorder) noexcept=0
Set the ErrorRecorder for this interface.
Definition: NvInferRuntimeCommon.h:194
virtual bool parse(const char *file, nvinfer1::INetworkDefinition &network, nvinfer1::DataType weightsType=nvinfer1::DataType::kFLOAT) noexcept=0
Parse a UFF file.
The TensorRT UFF parser API namespace.
virtual bool registerOutput(const char *outputName) noexcept=0
Register an output name of a UFF network.
void shutdownProtobufLibrary(void) noexcept
Shuts down protocol buffers library.
virtual int32_t getUffRequiredVersionMajor() noexcept=0
Return Version Major of the UFF.
char field type. String for length>1.
DataType
The type of weights and tensors.
Definition: NvInferRuntimeCommon.h:155
nvinfer1::Dims field type.
virtual void setPluginNamespace(const char *libNamespace) noexcept=0
Set the namespace used to lookup and create plugins in the network.
Class used for parsing models described using the UFF format.
Definition: NvUffParser.h:133
Reference counted application-implemented error reporting interface for TensorRT objects.
Definition: NvInferRuntimeCommon.h:1373
UffInputOrder
The different possible supported input order.
Definition: NvUffParser.h:78
virtual bool parseBuffer(const char *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.
32-bit floating point format.
IUffParser * createUffParser() noexcept
Creates a IUffParser object.
virtual TRT_DEPRECATED void destroy() noexcept=0
nvinfer1::DataType field type.
virtual bool registerInput(const char *inputName, nvinfer1::Dims inputDims, UffInputOrder inputOrder) noexcept=0
Register an input name of a UFF network with the associated Dimensions.
#define TRT_DEPRECATED
< Items that are marked as deprecated will be removed in a future release.
Definition: NvInferRuntimeCommon.h:76
virtual nvinfer1::IErrorRecorder * getErrorRecorder() const noexcept=0
get the ErrorRecorder assigned to this interface.
virtual int32_t getUffRequiredVersionPatch() noexcept=0
Return Patch Version of the UFF.
Copyright © 2024 NVIDIA Corporation
Privacy Policy |
Manage My Privacy |
Do Not Sell or Share My Data |
Terms of Service |
Accessibility |
Corporate Policies |
Product Security |
Contact