nvcaffeparser1 Namespace Reference

The TensorRT Caffe parser API namespace. More...

Classes
class	BinaryProtoBlob
class	BlobNameToTensor
class	CaffeParser
class	CaffeWeightFactory
class	IBinaryProtoBlob
	Object used to store and query data extracted from a binaryproto file using the ICaffeParser. More...
class	IBlobNameToTensor
	Object used to store and query Tensors after they have been extracted from a Caffe model using the ICaffeParser. More...
class	ICaffeParser
	Class used for parsing Caffe models. More...
class	IPluginFactory
	Plugin factory used to configure plugins. More...
class	IPluginFactoryExt
	Plugin factory used to configure plugins with added support for TRT versioning. More...
class	IPluginFactoryV2
	Plugin factory used to configure plugins. More...

Typedefs
typedef nvinfer1::ILayer *(*	LayerParseFn) (nvinfer1::INetworkDefinition &, const trtcaffe::LayerParameter &, CaffeWeightFactory &, BlobNameToTensor &)

Enumerations
enum	WeightType {   WeightType::kGENERIC = 0,   WeightType::kBIAS = 1,   WeightType::kMEAN = 0,   WeightType::kVARIANCE = 1,   WeightType::kMOVING_AVERAGE = 2,   WeightType::kNVMEAN = 0,   WeightType::kNVVARIANCE = 1,   WeightType::kNVSCALE = 3,   WeightType::kNVBIAS = 4 }

Functions
ICaffeParser *	createCaffeParser ()
	Creates a ICaffeParser object. More...
void	shutdownProtobufLibrary ()
	Shuts down protocol buffers library. More...
bool	checkBlobs (const trtcaffe::LayerParameter &msg, int bottoms, int tops)
nvinfer1::ILayer *	parseAbsVal (nvinfer1::INetworkDefinition &network, const trtcaffe::LayerParameter &msg, CaffeWeightFactory &, BlobNameToTensor &tensors)
nvinfer1::ILayer *	parseBatchNormalization (nvinfer1::INetworkDefinition &network, const trtcaffe::LayerParameter &msg, CaffeWeightFactory &weightFactory, BlobNameToTensor &tensors)
nvinfer1::ILayer *	parseBNLL (nvinfer1::INetworkDefinition &network, const trtcaffe::LayerParameter &msg, CaffeWeightFactory &, BlobNameToTensor &tensors)
nvinfer1::ILayer *	parseClip (nvinfer1::INetworkDefinition &network, const trtcaffe::LayerParameter &msg, CaffeWeightFactory &, BlobNameToTensor &tensors)
nvinfer1::ILayer *	parseConcat (nvinfer1::INetworkDefinition &network, const trtcaffe::LayerParameter &msg, CaffeWeightFactory &, BlobNameToTensor &tensors)
nvinfer1::ILayer *	parseConvolution (nvinfer1::INetworkDefinition &network, const trtcaffe::LayerParameter &msg, CaffeWeightFactory &weightFactory, BlobNameToTensor &tensors)
nvinfer1::ILayer *	parseCrop (nvinfer1::INetworkDefinition &network, const trtcaffe::LayerParameter &msg, CaffeWeightFactory &, BlobNameToTensor &tensors)
nvinfer1::ILayer *	parseDeconvolution (nvinfer1::INetworkDefinition &network, const trtcaffe::LayerParameter &msg, CaffeWeightFactory &weightFactory, BlobNameToTensor &tensors)
nvinfer1::ILayer *	parseEltwise (nvinfer1::INetworkDefinition &network, const trtcaffe::LayerParameter &msg, CaffeWeightFactory &, BlobNameToTensor &tensors)
nvinfer1::ILayer *	parseELU (nvinfer1::INetworkDefinition &network, const trtcaffe::LayerParameter &msg, CaffeWeightFactory &, BlobNameToTensor &tensors)
nvinfer1::ILayer *	parseInnerProduct (nvinfer1::INetworkDefinition &network, const trtcaffe::LayerParameter &msg, CaffeWeightFactory &weightFactory, BlobNameToTensor &tensors)
nvinfer1::ILayer *	parseLRN (nvinfer1::INetworkDefinition &network, const trtcaffe::LayerParameter &msg, CaffeWeightFactory &, BlobNameToTensor &tensors)
nvinfer1::ILayer *	parsePermute (nvinfer1::INetworkDefinition &network, const trtcaffe::LayerParameter &msg, CaffeWeightFactory &, BlobNameToTensor &tensors)
nvinfer1::ILayer *	parsePooling (nvinfer1::INetworkDefinition &network, const trtcaffe::LayerParameter &msg, CaffeWeightFactory &, BlobNameToTensor &tensors)
nvinfer1::ILayer *	parsePower (nvinfer1::INetworkDefinition &network, const trtcaffe::LayerParameter &msg, CaffeWeightFactory &weightFactory, BlobNameToTensor &tensors)
nvinfer1::ILayer *	parsePReLU (nvinfer1::INetworkDefinition &network, const trtcaffe::LayerParameter &msg, CaffeWeightFactory &weightFactory, BlobNameToTensor &tensors)
nvinfer1::ILayer *	parseReduction (nvinfer1::INetworkDefinition &network, const trtcaffe::LayerParameter &msg, CaffeWeightFactory &weightFactory, BlobNameToTensor &tensors)
nvinfer1::ILayer *	parseReLU (nvinfer1::INetworkDefinition &network, const trtcaffe::LayerParameter &msg, CaffeWeightFactory &, BlobNameToTensor &tensors)
nvinfer1::ILayer *	parseReshape (nvinfer1::INetworkDefinition &network, const trtcaffe::LayerParameter &msg, CaffeWeightFactory &, BlobNameToTensor &tensors)
nvinfer1::ILayer *	parseScale (nvinfer1::INetworkDefinition &network, const trtcaffe::LayerParameter &msg, CaffeWeightFactory &weightFactory, BlobNameToTensor &tensors)
nvinfer1::ILayer *	parseSigmoid (nvinfer1::INetworkDefinition &network, const trtcaffe::LayerParameter &msg, CaffeWeightFactory &, BlobNameToTensor &tensors)
nvinfer1::ILayer *	parseSoftMax (nvinfer1::INetworkDefinition &network, const trtcaffe::LayerParameter &msg, CaffeWeightFactory &, BlobNameToTensor &tensors)
nvinfer1::ILayer *	parseTanH (nvinfer1::INetworkDefinition &network, const trtcaffe::LayerParameter &msg, CaffeWeightFactory &, BlobNameToTensor &tensors)
template<typename T >
bool	bnConvertWrap (float scaleFactor, const Weights &variance, const Weights &mean, const Weights &scaleBlob, const Weights &biasBlob, Weights &shift, Weights &scale, float eps, bool nvCaffe, CaffeWeightFactory &weightFactory)
bool	readBinaryProto (trtcaffe::NetParameter *net, const char *file, size_t bufSize)
bool	readTextProto (trtcaffe::NetParameter *net, const char *file)

Variables
static std::unordered_map< std::string, LayerParseFn >	gParseTable

Detailed Description

The TensorRT Caffe parser API namespace.

Typedef Documentation

LayerParseFn

typedef nvinfer1::ILayer*(* nvcaffeparser1::LayerParseFn) (nvinfer1::INetworkDefinition &, const trtcaffe::LayerParameter &, CaffeWeightFactory &, BlobNameToTensor &)

Enumeration Type Documentation

WeightType

enum nvcaffeparser1::WeightType

strong

Enumerator
kGENERIC
kBIAS
kMEAN
kVARIANCE
kMOVING_AVERAGE
kNVMEAN
kNVVARIANCE
kNVSCALE
kNVBIAS

Function Documentation

createCaffeParser()

ICaffeParser * nvcaffeparser1::createCaffeParser

(

)

Creates a ICaffeParser object.

Returns

A pointer to the ICaffeParser object is returned.

See also

nvcaffeparser1::ICaffeParser

Here is the caller graph for this function:

shutdownProtobufLibrary()

void nvcaffeparser1::shutdownProtobufLibrary

(

)

Shuts down protocol buffers library.

Note

No part of the protocol buffers library can be used after this function is called.

Here is the caller graph for this function:

checkBlobs()

bool nvcaffeparser1::checkBlobs ( const trtcaffe::LayerParameter &  msg, int  bottoms, int  tops  )

inline

Here is the caller graph for this function:

parseAbsVal()

ILayer * nvcaffeparser1::parseAbsVal	(	nvinfer1::INetworkDefinition &	network,
		const trtcaffe::LayerParameter &	msg,
		CaffeWeightFactory &	,
		BlobNameToTensor &	tensors
	)

Here is the call graph for this function:

Here is the caller graph for this function:

parseBatchNormalization()

ILayer * nvcaffeparser1::parseBatchNormalization	(	nvinfer1::INetworkDefinition &	network,
		const trtcaffe::LayerParameter &	msg,
		CaffeWeightFactory &	weightFactory,
		BlobNameToTensor &	tensors
	)

Here is the call graph for this function:

Here is the caller graph for this function:

parseBNLL()

ILayer * nvcaffeparser1::parseBNLL	(	nvinfer1::INetworkDefinition &	network,
		const trtcaffe::LayerParameter &	msg,
		CaffeWeightFactory &	,
		BlobNameToTensor &	tensors
	)

Here is the call graph for this function:

Here is the caller graph for this function:

parseClip()

ILayer * nvcaffeparser1::parseClip	(	nvinfer1::INetworkDefinition &	network,
		const trtcaffe::LayerParameter &	msg,
		CaffeWeightFactory &	,
		BlobNameToTensor &	tensors
	)

Here is the call graph for this function:

Here is the caller graph for this function:

parseConcat()

ILayer * nvcaffeparser1::parseConcat	(	nvinfer1::INetworkDefinition &	network,
		const trtcaffe::LayerParameter &	msg,
		CaffeWeightFactory &	,
		BlobNameToTensor &	tensors
	)

Here is the call graph for this function:

Here is the caller graph for this function:

parseConvolution()

ILayer * nvcaffeparser1::parseConvolution	(	nvinfer1::INetworkDefinition &	network,
		const trtcaffe::LayerParameter &	msg,
		CaffeWeightFactory &	weightFactory,
		BlobNameToTensor &	tensors
	)

Here is the call graph for this function:

Here is the caller graph for this function:

parseCrop()

ILayer * nvcaffeparser1::parseCrop	(	nvinfer1::INetworkDefinition &	network,
		const trtcaffe::LayerParameter &	msg,
		CaffeWeightFactory &	,
		BlobNameToTensor &	tensors
	)

Here is the call graph for this function:

Here is the caller graph for this function:

parseDeconvolution()

ILayer * nvcaffeparser1::parseDeconvolution	(	nvinfer1::INetworkDefinition &	network,
		const trtcaffe::LayerParameter &	msg,
		CaffeWeightFactory &	weightFactory,
		BlobNameToTensor &	tensors
	)

Here is the call graph for this function:

Here is the caller graph for this function:

parseEltwise()

ILayer * nvcaffeparser1::parseEltwise	(	nvinfer1::INetworkDefinition &	network,
		const trtcaffe::LayerParameter &	msg,
		CaffeWeightFactory &	,
		BlobNameToTensor &	tensors
	)

Here is the call graph for this function:

Here is the caller graph for this function:

parseELU()

ILayer * nvcaffeparser1::parseELU	(	nvinfer1::INetworkDefinition &	network,
		const trtcaffe::LayerParameter &	msg,
		CaffeWeightFactory &	,
		BlobNameToTensor &	tensors
	)

Here is the call graph for this function:

Here is the caller graph for this function:

parseInnerProduct()

ILayer * nvcaffeparser1::parseInnerProduct	(	nvinfer1::INetworkDefinition &	network,
		const trtcaffe::LayerParameter &	msg,
		CaffeWeightFactory &	weightFactory,
		BlobNameToTensor &	tensors
	)

Here is the call graph for this function:

Here is the caller graph for this function:

parseLRN()

ILayer * nvcaffeparser1::parseLRN	(	nvinfer1::INetworkDefinition &	network,
		const trtcaffe::LayerParameter &	msg,
		CaffeWeightFactory &	,
		BlobNameToTensor &	tensors
	)

Here is the call graph for this function:

Here is the caller graph for this function:

parsePermute()

ILayer * nvcaffeparser1::parsePermute	(	nvinfer1::INetworkDefinition &	network,
		const trtcaffe::LayerParameter &	msg,
		CaffeWeightFactory &	,
		BlobNameToTensor &	tensors
	)

Here is the call graph for this function:

Here is the caller graph for this function:

parsePooling()

ILayer * nvcaffeparser1::parsePooling	(	nvinfer1::INetworkDefinition &	network,
		const trtcaffe::LayerParameter &	msg,
		CaffeWeightFactory &	,
		BlobNameToTensor &	tensors
	)

Here is the call graph for this function:

Here is the caller graph for this function:

parsePower()

ILayer * nvcaffeparser1::parsePower	(	nvinfer1::INetworkDefinition &	network,
		const trtcaffe::LayerParameter &	msg,
		CaffeWeightFactory &	weightFactory,
		BlobNameToTensor &	tensors
	)

Here is the call graph for this function:

Here is the caller graph for this function:

parsePReLU()

ILayer * nvcaffeparser1::parsePReLU	(	nvinfer1::INetworkDefinition &	network,
		const trtcaffe::LayerParameter &	msg,
		CaffeWeightFactory &	weightFactory,
		BlobNameToTensor &	tensors
	)

Here is the call graph for this function:

Here is the caller graph for this function:

parseReduction()

ILayer * nvcaffeparser1::parseReduction	(	nvinfer1::INetworkDefinition &	network,
		const trtcaffe::LayerParameter &	msg,
		CaffeWeightFactory &	weightFactory,
		BlobNameToTensor &	tensors
	)

Here is the call graph for this function:

Here is the caller graph for this function:

parseReLU()

ILayer * nvcaffeparser1::parseReLU	(	nvinfer1::INetworkDefinition &	network,
		const trtcaffe::LayerParameter &	msg,
		CaffeWeightFactory &	,
		BlobNameToTensor &	tensors
	)

Here is the call graph for this function:

Here is the caller graph for this function:

parseReshape()

ILayer * nvcaffeparser1::parseReshape	(	nvinfer1::INetworkDefinition &	network,
		const trtcaffe::LayerParameter &	msg,
		CaffeWeightFactory &	,
		BlobNameToTensor &	tensors
	)

Here is the call graph for this function:

Here is the caller graph for this function:

parseScale()

ILayer * nvcaffeparser1::parseScale	(	nvinfer1::INetworkDefinition &	network,
		const trtcaffe::LayerParameter &	msg,
		CaffeWeightFactory &	weightFactory,
		BlobNameToTensor &	tensors
	)

Here is the call graph for this function:

Here is the caller graph for this function:

parseSigmoid()

ILayer * nvcaffeparser1::parseSigmoid	(	nvinfer1::INetworkDefinition &	network,
		const trtcaffe::LayerParameter &	msg,
		CaffeWeightFactory &	,
		BlobNameToTensor &	tensors
	)

Here is the call graph for this function:

Here is the caller graph for this function:

parseSoftMax()

ILayer * nvcaffeparser1::parseSoftMax	(	nvinfer1::INetworkDefinition &	network,
		const trtcaffe::LayerParameter &	msg,
		CaffeWeightFactory &	,
		BlobNameToTensor &	tensors
	)

Here is the call graph for this function:

Here is the caller graph for this function:

parseTanH()

ILayer * nvcaffeparser1::parseTanH	(	nvinfer1::INetworkDefinition &	network,
		const trtcaffe::LayerParameter &	msg,
		CaffeWeightFactory &	,
		BlobNameToTensor &	tensors
	)

Here is the call graph for this function:

Here is the caller graph for this function:

bnConvertWrap()

template<typename T >

bool nvcaffeparser1::bnConvertWrap ( float  scaleFactor, const Weights &  variance, const Weights &  mean, const Weights &  scaleBlob, const Weights &  biasBlob, Weights &  shift, Weights &  scale, float  eps, bool  nvCaffe, CaffeWeightFactory &  weightFactory  )

inline

Here is the call graph for this function:

Here is the caller graph for this function:

readBinaryProto()

bool nvcaffeparser1::readBinaryProto	(	trtcaffe::NetParameter *	net,
		const char *	file,
		size_t	bufSize
	)

Here is the caller graph for this function:

readTextProto()

bool nvcaffeparser1::readTextProto	(	trtcaffe::NetParameter *	net,
		const char *	file
	)

Here is the caller graph for this function:

Variable Documentation

gParseTable

std::unordered_map<std::string, LayerParseFn> nvcaffeparser1::gParseTable

static

Initial value:

{
    {"Convolution", parseConvolution},
    {"Pooling", parsePooling},
    {"InnerProduct", parseInnerProduct},
    {"ReLU", parseReLU},
    {"Softmax", parseSoftMax},
    {"SoftmaxWithLoss", parseSoftMax},
    {"LRN", parseLRN},
    {"Power", parsePower},
    {"Eltwise", parseEltwise},
    {"Concat", parseConcat},
    {"Deconvolution", parseDeconvolution},
    {"Sigmoid", parseSigmoid},
    {"TanH", parseTanH},
    {"BatchNorm", parseBatchNormalization},
    {"Scale", parseScale},
    {"Crop", parseCrop},
    {"Reduction", parseReduction},
    {"Reshape", parseReshape},
    {"Permute", parsePermute},
    {"ELU", parseELU},
    {"BNLL", parseBNLL},
    {"Clip", parseClip},
    {"AbsVal", parseAbsVal},
    {"PReLU", parsePReLU}
}