Go to the documentation of this file.
50 #ifndef NV_CAFFE_PARSER_H
51 #define NV_CAFFE_PARSER_H
107 virtual const void* getData() noexcept = 0;
132 virtual bool isPluginV2(
const char* layerName) noexcept = 0;
143 int32_t nbWeights,
const char* libNamespace =
"") noexcept
297 extern "C" TENSORRTAPI
void* createNvCaffeParser_INTERNAL() noexcept;
Class used for parsing Caffe models.
Definition: NvCaffeParser.h:157
virtual TRT_DEPRECATED void destroy() noexcept=0
virtual void setPluginFactoryV2(IPluginFactoryV2 *factory) noexcept=0
Set the IPluginFactoryV2 used to create the user defined pluginV2 objects.
A network definition for input to the builder.
Definition: NvInfer.h:5450
An array of weights used as a layer parameter.
Definition: NvInferRuntime.h:152
virtual nvinfer1::ITensor * find(const char *name) const noexcept=0
Given a blob name, returns a pointer to a ITensor object.
virtual void setPluginNamespace(const char *libNamespace) noexcept=0
Set the namespace used to lookup and create plugins in the network.
void shutdownProtobufLibrary() noexcept
Shuts down protocol buffers library.
ICaffeParser * createCaffeParser() noexcept
Creates a ICaffeParser object.
virtual void setProtobufBufferSize(size_t size) noexcept=0
Set buffer size for the parsing and storage of the learned model.
Plugin class for user-implemented layers.
Definition: NvInferRuntimeCommon.h:414
virtual void setErrorRecorder(nvinfer1::IErrorRecorder *recorder) noexcept=0
Set the ErrorRecorder for this interface.
virtual nvinfer1::IErrorRecorder * getErrorRecorder() const noexcept=0
get the ErrorRecorder assigned to this interface.
virtual IBinaryProtoBlob * parseBinaryProto(const char *fileName) noexcept=0
Parse and extract data stored in binaryproto file.
virtual bool isPluginV2(const char *layerName) noexcept=0
A user implemented function that determines if a layer configuration is provided by an IPluginV2.
virtual const IBlobNameToTensor * parseBuffers(const char *deployBuffer, std::size_t deployLength, const char *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...
The TensorRT Caffe parser API namespace.
Plugin factory used to configure plugins.
Definition: NvCaffeParser.h:124
DataType
The type of weights and tensors.
Definition: NvInferRuntimeCommon.h:155
A tensor in a network definition.
Definition: NvInfer.h:187
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...
Reference counted application-implemented error reporting interface for TensorRT objects.
Definition: NvInferRuntimeCommon.h:1373
Object used to store and query Tensors after they have been extracted from a Caffe model using the IC...
Definition: NvCaffeParser.h:80
virtual nvinfer1::IPluginV2 * createPlugin(const char *layerName, const nvinfer1::Weights *weights, int32_t nbWeights, const char *libNamespace="") noexcept=0
Creates a plugin.
Descriptor for four-dimensional data.
Definition: NvInferLegacyDims.h:216
Object used to store and query data extracted from a binaryproto file using the ICaffeParser.
Definition: NvCaffeParser.h:104
#define TRT_DEPRECATED
< Items that are marked as deprecated will be removed in a future release.
Definition: NvInferRuntimeCommon.h:76
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