50#ifndef NV_CAFFE_PARSER_H
51#define NV_CAFFE_PARSER_H
107 virtual const void* getData()
noexcept = 0;
143 int32_t nbWeights,
const char* libNamespace =
"") noexcept
239 virtual
void setPluginNamespace(const
char* libNamespace) noexcept = 0;
258 virtual
void setErrorRecorder(
nvinfer1::IErrorRecorder* recorder) noexcept = 0;
270 virtual
nvinfer1::IErrorRecorder* getErrorRecorder() const noexcept = 0;
297extern "C" TENSORRTAPI
void* createNvCaffeParser_INTERNAL() noexcept;
#define TRT_DEPRECATED
< Items that are marked as deprecated will be removed in a future release.
Definition: NvInferRuntimeCommon.h:77
Object used to store and query data extracted from a binaryproto file using the ICaffeParser.
Definition: NvCaffeParser.h:105
virtual TRT_DEPRECATED void destroy() noexcept=0
Object used to store and query Tensors after they have been extracted from a Caffe model using the IC...
Definition: NvCaffeParser.h:81
virtual nvinfer1::ITensor * find(const char *name) const noexcept=0
Given a blob name, returns a pointer to a ITensor object.
Class used for parsing Caffe models.
Definition: NvCaffeParser.h:158
virtual const IBlobNameToTensor * parseBuffers(const uint8_t *deployBuffer, std::size_t deployLength, const uint8_t *modelBuffer, std::size_t modelLength, nvinfer1::INetworkDefinition &network, nvinfer1::DataType weightType) noexcept=0
Parse a deploy prototxt and a binaryproto Caffe model from memory buffers to extract network definiti...
virtual void setProtobufBufferSize(size_t size) noexcept=0
Set buffer size for the parsing and storage of the learned model.
virtual IBinaryProtoBlob * parseBinaryProto(const char *fileName) noexcept=0
Parse and extract data stored in binaryproto file.
virtual TRT_DEPRECATED void destroy() noexcept=0
Destroy this ICaffeParser object.
virtual const IBlobNameToTensor * parse(const char *deploy, const char *model, nvinfer1::INetworkDefinition &network, nvinfer1::DataType weightType) noexcept=0
Parse a prototxt file and a binaryproto Caffe model to extract network definition and weights associa...
Plugin factory used to configure plugins.
Definition: NvCaffeParser.h:125
virtual nvinfer1::IPluginV2 * createPlugin(const char *layerName, const nvinfer1::Weights *weights, int32_t nbWeights, const char *libNamespace="") noexcept=0
Creates a plugin.
virtual bool isPluginV2(const char *layerName) noexcept=0
A user implemented function that determines if a layer configuration is provided by an IPluginV2.
Descriptor for four-dimensional data.
Definition: NvInferLegacyDims.h:196
A network definition for input to the builder.
Definition: NvInfer.h:6030
Plugin class for user-implemented layers.
Definition: NvInferRuntimeCommon.h:411
A tensor in a network definition.
Definition: NvInfer.h:196
An array of weights used as a layer parameter.
Definition: NvInferRuntime.h:157
The TensorRT Caffe parser API namespace.
void shutdownProtobufLibrary() noexcept
Shuts down protocol buffers library.
ICaffeParser * createCaffeParser() noexcept
Creates a ICaffeParser object.
The TensorRT API version 1 namespace.
DataType
The type of weights and tensors.
Definition: NvInferRuntimeCommon.h:151