nvcaffeparser1::BlobNameToTensor Class Reference

Inheritance diagram for nvcaffeparser1::BlobNameToTensor:

Collaboration diagram for nvcaffeparser1::BlobNameToTensor:

Public Member Functions
void	add (const std::string &name, nvinfer1::ITensor *tensor)
nvinfer1::ITensor *	find (const char *name) const override
	Given a blob name, returns a pointer to a ITensor object. More...
nvinfer1::ITensor *&	operator[] (const std::string &name)
void	setTensorNames ()
	~BlobNameToTensor () override=default
bool	isOK ()

Private Attributes
std::map< std::string, nvinfer1::ITensor * >	mMap
bool	mError {false}

Constructor & Destructor Documentation

~BlobNameToTensor()

nvcaffeparser1::BlobNameToTensor::~BlobNameToTensor ( )

overridedefault

Member Function Documentation

add()

void nvcaffeparser1::BlobNameToTensor::add ( const std::string &  name, nvinfer1::ITensor *  tensor  )

inline

find()

nvinfer1::ITensor* nvcaffeparser1::BlobNameToTensor::find ( const char *  name ) const

inlineoverridevirtual

Given a blob name, returns a pointer to a ITensor object.

Parameters

name	Caffe blob name for which the user wants the corresponding ITensor.

Returns

ITensor* corresponding to the queried name. If no such ITensor exists, then nullptr is returned.

Implements nvcaffeparser1::IBlobNameToTensor.

operator[]()

nvinfer1::ITensor*& nvcaffeparser1::BlobNameToTensor::operator[] ( const std::string &  name )

inline

setTensorNames()

void nvcaffeparser1::BlobNameToTensor::setTensorNames ( )

inline

isOK()

bool nvcaffeparser1::BlobNameToTensor::isOK ( )

inline

Member Data Documentation

mMap

std::map<std::string, nvinfer1::ITensor*> nvcaffeparser1::BlobNameToTensor::mMap

private

mError

bool nvcaffeparser1::BlobNameToTensor::mError {false}

private

---

The documentation for this class was generated from the following file:

• blobNameToTensor.h