50 #ifndef NV_CAFFE_PARSER_H 51 #define NV_CAFFE_PARSER_H 101 virtual const void* getData() TRTNOEXCEPT = 0;
104 virtual void destroy() TRTNOEXCEPT = 0;
123 virtual bool isPlugin(
const char* layerName) TRTNOEXCEPT = 0;
145 virtual int getVersion()
const TRTNOEXCEPT
147 return NV_TENSORRT_VERSION;
155 virtual bool isPluginExt(
const char* layerName) TRTNOEXCEPT = 0;
171 virtual bool isPluginV2(
const char* layerName) TRTNOEXCEPT = 0;
231 std::size_t deployLength,
232 const char* modelBuffer,
233 std::size_t modelLength,
249 virtual IBinaryProtoBlob* parseBinaryProto(
const char* fileName) TRTNOEXCEPT = 0;
258 virtual void setProtobufBufferSize(
size_t size) TRTNOEXCEPT = 0;
265 virtual void setPluginFactory(
IPluginFactory* factory) TRTNOEXCEPT = 0;
277 virtual void destroy() TRTNOEXCEPT = 0;
289 virtual void setPluginNamespace(
const char* libNamespace) TRTNOEXCEPT = 0;
343 extern "C" TENSORRTAPI
void* createNvCaffeParser_INTERNAL();
DataType
The type of weights and tensors.
Definition: NvInferRuntimeCommon.h:162
Plugin class for user-implemented layers.
Definition: NvInferRuntimeCommon.h:344
virtual nvinfer1::ITensor * find(const char *name) const =0
Given a blob name, returns a pointer to a ITensor object.
Plugin factory used to configure plugins.
Definition: NvCaffeParser.h:163
Plugin class for user-implemented layers.
Definition: NvInferRuntime.h:134
void shutdownProtobufLibrary()
Shuts down protocol buffers library.
Class used for parsing Caffe models.
Definition: NvCaffeParser.h:194
Object used to store and query data extracted from a binaryproto file using the ICaffeParser.
Definition: NvCaffeParser.h:98
A network definition for input to the builder.
Definition: NvInfer.h:4717
The TensorRT Caffe parser API namespace.
Plugin factory used to configure plugins with added support for TRT versioning.
Definition: NvCaffeParser.h:142
Descriptor for data with one index dimension, one channel dimension and two spatial dimensions...
Definition: NvInfer.h:316
Reference counted application-implemented error reporting interface for TensorRT objects.
Definition: NvInferRuntimeCommon.h:1142
Plugin factory used to configure plugins.
Definition: NvCaffeParser.h:115
A tensor in a network definition.
Definition: NvInfer.h:463
An array of weights used as a layer parameter.
Definition: NvInferRuntime.h:98
ICaffeParser * createCaffeParser()
Creates a ICaffeParser object.
Object used to store and query Tensors after they have been extracted from a Caffe model using the IC...
Definition: NvCaffeParser.h:74