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

The SampleMLPParams structure groups the additional parameters required by the MLP sample. More...

Inheritance diagram for SampleMLPParams:
Collaboration diagram for SampleMLPParams:

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...
 
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 SampleMLPParams structure groups the additional parameters required by the MLP sample.

Member Data Documentation

◆ inputH

int SampleMLPParams::inputH

The input height.

◆ inputW

int SampleMLPParams::inputW

The input width.

◆ outputSize

int SampleMLPParams::outputSize

The output size.

◆ weightsFile

std::string SampleMLPParams::weightsFile

The filename of the weights file.

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