TensorRT  7.2.1.6
NVIDIA TensorRT
Looking for a C++ dev who knows TensorRT?
I'm looking for work. Hire me!
All Classes Namespaces Functions Variables Typedefs Enumerations Enumerator Friends Pages
EntropyCalibratorImpl< TBatchStream > Class Template Reference

Implements common functionality for Entropy calibrators. More...

Collaboration diagram for EntropyCalibratorImpl< TBatchStream >:

Public Member Functions

 EntropyCalibratorImpl (TBatchStream stream, int firstBatch, std::string networkName, const char *inputBlobName, bool readCache=true)
 
virtual ~EntropyCalibratorImpl ()
 
int getBatchSize () const
 
bool getBatch (void *bindings[], const char *names[], int nbBindings)
 
const void * readCalibrationCache (size_t &length)
 
void writeCalibrationCache (const void *cache, size_t length)
 
 EntropyCalibratorImpl (BatchStream &stream, int firstBatch, const std::string &networkName, const char *inputBlobName, bool readCache=true)
 
virtual ~EntropyCalibratorImpl ()
 
int getBatchSize () const
 
bool getBatch (void *bindings[], const char *names[], int nbBindings)
 
const void * readCalibrationCache (size_t &length)
 
void writeCalibrationCache (const void *cache, size_t length)
 

Private Attributes

TBatchStream mStream
 
size_t mInputCount
 
std::string mCalibrationTableName
 
const char * mInputBlobName
 
bool mReadCache {true}
 
void * mDeviceInput {nullptr}
 
std::vector< char > mCalibrationCache
 
BatchStream mStream
 

Detailed Description

template<typename TBatchStream>
class EntropyCalibratorImpl< TBatchStream >

Implements common functionality for Entropy calibrators.

Constructor & Destructor Documentation

◆ EntropyCalibratorImpl() [1/2]

template<typename TBatchStream >
EntropyCalibratorImpl< TBatchStream >::EntropyCalibratorImpl ( TBatchStream  stream,
int  firstBatch,
std::string  networkName,
const char *  inputBlobName,
bool  readCache = true 
)
inline

◆ ~EntropyCalibratorImpl() [1/2]

template<typename TBatchStream >
virtual EntropyCalibratorImpl< TBatchStream >::~EntropyCalibratorImpl ( )
inlinevirtual

◆ EntropyCalibratorImpl() [2/2]

template<typename TBatchStream >
EntropyCalibratorImpl< TBatchStream >::EntropyCalibratorImpl ( BatchStream stream,
int  firstBatch,
const std::string &  networkName,
const char *  inputBlobName,
bool  readCache = true 
)
inline
Here is the call graph for this function:

◆ ~EntropyCalibratorImpl() [2/2]

template<typename TBatchStream >
virtual EntropyCalibratorImpl< TBatchStream >::~EntropyCalibratorImpl ( )
inlinevirtual

Member Function Documentation

◆ getBatchSize() [1/2]

template<typename TBatchStream >
int EntropyCalibratorImpl< TBatchStream >::getBatchSize ( ) const
inline

◆ getBatch() [1/2]

template<typename TBatchStream >
bool EntropyCalibratorImpl< TBatchStream >::getBatch ( void *  bindings[],
const char *  names[],
int  nbBindings 
)
inline

◆ readCalibrationCache() [1/2]

template<typename TBatchStream >
const void* EntropyCalibratorImpl< TBatchStream >::readCalibrationCache ( size_t &  length)
inline

◆ writeCalibrationCache() [1/2]

template<typename TBatchStream >
void EntropyCalibratorImpl< TBatchStream >::writeCalibrationCache ( const void *  cache,
size_t  length 
)
inline

◆ getBatchSize() [2/2]

template<typename TBatchStream >
int EntropyCalibratorImpl< TBatchStream >::getBatchSize ( ) const
inline

◆ getBatch() [2/2]

template<typename TBatchStream >
bool EntropyCalibratorImpl< TBatchStream >::getBatch ( void *  bindings[],
const char *  names[],
int  nbBindings 
)
inline

◆ readCalibrationCache() [2/2]

template<typename TBatchStream >
const void* EntropyCalibratorImpl< TBatchStream >::readCalibrationCache ( size_t &  length)
inline

◆ writeCalibrationCache() [2/2]

template<typename TBatchStream >
void EntropyCalibratorImpl< TBatchStream >::writeCalibrationCache ( const void *  cache,
size_t  length 
)
inline

Member Data Documentation

◆ mStream [1/2]

template<typename TBatchStream >
TBatchStream EntropyCalibratorImpl< TBatchStream >::mStream
private

◆ mInputCount

template<typename TBatchStream >
size_t EntropyCalibratorImpl< TBatchStream >::mInputCount
private

◆ mCalibrationTableName

template<typename TBatchStream >
std::string EntropyCalibratorImpl< TBatchStream >::mCalibrationTableName
private

◆ mInputBlobName

template<typename TBatchStream >
const char * EntropyCalibratorImpl< TBatchStream >::mInputBlobName
private

◆ mReadCache

template<typename TBatchStream >
bool EntropyCalibratorImpl< TBatchStream >::mReadCache {true}
private

◆ mDeviceInput

template<typename TBatchStream >
void * EntropyCalibratorImpl< TBatchStream >::mDeviceInput {nullptr}
private

◆ mCalibrationCache

template<typename TBatchStream >
std::vector< char > EntropyCalibratorImpl< TBatchStream >::mCalibrationCache
private

◆ mStream [2/2]

template<typename TBatchStream >
BatchStream EntropyCalibratorImpl< TBatchStream >::mStream
private

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