50 #ifndef NV_UFF_PARSER_H 51 #define NV_UFF_PARSER_H 56 #define UFF_REQUIRED_VERSION_MAJOR 0 57 #define UFF_REQUIRED_VERSION_MINOR 6 58 #define UFF_REQUIRED_VERSION_PATCH 5 111 FieldMap(
const char* name,
const void* data,
const FieldType type,
int length = 1) TRTNOEXCEPT;
133 virtual bool isPlugin(
const char* layerName) TRTNOEXCEPT = 0;
159 virtual int getVersion()
const TRTNOEXCEPT
161 return NV_TENSORRT_VERSION;
169 virtual bool isPluginExt(
const char* layerName) TRTNOEXCEPT = 0;
196 virtual bool registerOutput(
const char* outputName) TRTNOEXCEPT = 0;
205 virtual bool parse(
const char* file,
217 virtual bool parseBuffer(
const char* buffer, std::size_t size,
221 virtual void destroy() TRTNOEXCEPT = 0;
226 virtual int getUffRequiredVersionMajor() TRTNOEXCEPT = 0;
231 virtual int getUffRequiredVersionMinor() TRTNOEXCEPT = 0;
236 virtual int getUffRequiredVersionPatch() TRTNOEXCEPT = 0;
243 virtual void setPluginFactory(
IPluginFactory* factory) TRTNOEXCEPT = 0;
255 virtual void setPluginNamespace(
const char* libNamespace) TRTNOEXCEPT = 0;
311 extern "C" TENSORRTAPI
void* createNvUffParser_INTERNAL() TRTNOEXCEPT;
void shutdownProtobufLibrary(void)
Shuts down protocol buffers library.
DataType
The type of weights and tensors.
Definition: NvInferRuntimeCommon.h:162
Class used for parsing models described using the UFF format.
Definition: NvUffParser.h:179
Definition: NvUffParser.h:114
Plugin class for user-implemented layers.
Definition: NvInferRuntime.h:134
An array of field params used as a layer parameter for plugin layers.
Definition: NvUffParser.h:103
FieldType
The possible field types for custom layer.
Definition: NvUffParser.h:84
nvinfer1::DataType field type.
The TensorRT UFF parser API namespace.
A network definition for input to the builder.
Definition: NvInfer.h:4717
nvinfer1::Dims field type.
Reference counted application-implemented error reporting interface for TensorRT objects.
Definition: NvInferRuntimeCommon.h:1142
Structure to define the dimensions of a tensor.
Definition: NvInferRuntimeCommon.h:206
IUffParser * createUffParser()
Creates a IUffParser object.
An array of weights used as a layer parameter.
Definition: NvInferRuntime.h:98
UffInputOrder
The different possible supported input order.
Definition: NvUffParser.h:72
char field type. String for length>1.
Plugin factory used to configure plugins with added support for TRT versioning.
Definition: NvUffParser.h:156
Plugin factory used to configure plugins.
Definition: NvUffParser.h:125