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
SampleINT8Params Struct Reference

The SampleINT8Params structure groups the additional parameters required by the INT8 sample. More...

Inheritance diagram for SampleINT8Params:
Collaboration diagram for SampleINT8Params:

Public Attributes

int nbCalBatches
 The number of batches for calibration. More...
 
int calBatchSize
 The calibration batch size. More...
 
std::string networkName
 The name of the network. More...
 
std::string prototxtFileName
 Filename of prototxt design file of a network. More...
 
std::string weightsFileName
 Filename of trained weights file of a network. More...
 
std::string meanFileName
 Filename of mean file of a network. More...
 
int32_t batchSize {1}
 Number of inputs in a batch. More...
 
int32_t dlaCore {-1}
 Specify the DLA core to run network on. More...
 
bool int8 {false}
 Allow runnning the network in Int8 mode. More...
 
bool fp16 {false}
 Allow running the network in FP16 mode. More...
 
std::vector< std::string > dataDirs
 Directory paths where sample data files are stored. More...
 
std::vector< std::string > inputTensorNames
 
std::vector< std::string > outputTensorNames
 

Detailed Description

The SampleINT8Params structure groups the additional parameters required by the INT8 sample.

Member Data Documentation

◆ nbCalBatches

int SampleINT8Params::nbCalBatches

The number of batches for calibration.

◆ calBatchSize

int SampleINT8Params::calBatchSize

The calibration batch size.

◆ networkName

std::string SampleINT8Params::networkName

The name of the network.

◆ prototxtFileName

std::string samplesCommon::CaffeSampleParams::prototxtFileName
inherited

Filename of prototxt design file of a network.

◆ weightsFileName

std::string samplesCommon::CaffeSampleParams::weightsFileName
inherited

Filename of trained weights file of a network.

◆ meanFileName

std::string samplesCommon::CaffeSampleParams::meanFileName
inherited

Filename of mean file of a network.

◆ batchSize

int32_t samplesCommon::SampleParams::batchSize {1}
inherited

Number of inputs in a batch.

◆ dlaCore

int32_t samplesCommon::SampleParams::dlaCore {-1}
inherited

Specify the DLA core to run network on.

◆ int8

bool samplesCommon::SampleParams::int8 {false}
inherited

Allow runnning the network in Int8 mode.

◆ fp16

bool samplesCommon::SampleParams::fp16 {false}
inherited

Allow running the network in FP16 mode.

◆ dataDirs

std::vector<std::string> samplesCommon::SampleParams::dataDirs
inherited

Directory paths where sample data files are stored.

◆ inputTensorNames

std::vector<std::string> samplesCommon::SampleParams::inputTensorNames
inherited

◆ outputTensorNames

std::vector<std::string> samplesCommon::SampleParams::outputTensorNames
inherited

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