TensorRT  7.2.1.6
NVIDIA TensorRT
Looking for a C++ dev who knows TensorRT?
I'm looking for work. Hire me!
samplesCommon::CaffeSampleParams Struct Reference

The CaffeSampleParams structure groups the additional parameters required by networks that use caffe. More...

Inheritance diagram for samplesCommon::CaffeSampleParams:
Collaboration diagram for samplesCommon::CaffeSampleParams:

Public Attributes

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 CaffeSampleParams structure groups the additional parameters required by networks that use caffe.

Member Data Documentation

◆ prototxtFileName

std::string samplesCommon::CaffeSampleParams::prototxtFileName

Filename of prototxt design file of a network.

◆ weightsFileName

std::string samplesCommon::CaffeSampleParams::weightsFileName

Filename of trained weights file of a network.

◆ meanFileName

std::string samplesCommon::CaffeSampleParams::meanFileName

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: