Implements common functionality for Entropy calibrators. More...
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 |
Implements common functionality for Entropy calibrators.
|
inline |
|
inlinevirtual |
|
inline |
|
inlinevirtual |
|
inline |
|
inline |
|
inline |
|
inline |
|
inline |
|
inline |
|
inline |
|
inline |
|
private |
|
private |
|
private |
|
private |
|
private |
|
private |
|
private |
|
private |