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

The SampleMNISTAPIParams structure groups the additional parameters required by the SampleMNISTAPI sample. More...

Inheritance diagram for SampleMNISTAPIParams:
Collaboration diagram for SampleMNISTAPIParams:

Public Attributes

int inputH
 The input height. More...
 
int inputW
 The input width. More...
 
int outputSize
 The output size. More...
 
std::string weightsFile
 The filename of the weights file. More...
 
std::string mnistMeansProto
 The proto file containing means. 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 SampleMNISTAPIParams structure groups the additional parameters required by the SampleMNISTAPI sample.

Member Data Documentation

◆ inputH

int SampleMNISTAPIParams::inputH

The input height.

◆ inputW

int SampleMNISTAPIParams::inputW

The input width.

◆ outputSize

int SampleMNISTAPIParams::outputSize

The output size.

◆ weightsFile

std::string SampleMNISTAPIParams::weightsFile

The filename of the weights file.

◆ mnistMeansProto

std::string SampleMNISTAPIParams::mnistMeansProto

The proto file containing means.

◆ 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: