SampleCharRNNParams Struct Reference

Inheritance diagram for SampleCharRNNParams:

Collaboration diagram for SampleCharRNNParams:

Public Attributes
int	layerCount
int	hiddenSize
int	seqSize
int	dataSize
int	vocabSize
int	outputSize
std::string	weightFileName
std::string	saveEngine
std::string	loadEngine
SampleCharRNNMaps	charMaps
SampleCharRNNWeightNames	weightNames
SampleCharRNNBindingNames	bindingNames
std::vector< std::string >	inputSentences
std::vector< std::string >	outputSentences
bool	useILoop
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

Member Data Documentation

layerCount

int SampleCharRNNParams::layerCount

hiddenSize

int SampleCharRNNParams::hiddenSize

seqSize

int SampleCharRNNParams::seqSize

dataSize

int SampleCharRNNParams::dataSize

vocabSize

int SampleCharRNNParams::vocabSize

outputSize

int SampleCharRNNParams::outputSize

weightFileName

std::string SampleCharRNNParams::weightFileName

saveEngine

std::string SampleCharRNNParams::saveEngine

loadEngine

std::string SampleCharRNNParams::loadEngine

charMaps

SampleCharRNNMaps SampleCharRNNParams::charMaps

weightNames

SampleCharRNNWeightNames SampleCharRNNParams::weightNames

bindingNames

SampleCharRNNBindingNames SampleCharRNNParams::bindingNames

inputSentences

std::vector<std::string> SampleCharRNNParams::inputSentences

outputSentences

std::vector<std::string> SampleCharRNNParams::outputSentences

useILoop

bool SampleCharRNNParams::useILoop

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:

• sampleCharRNN.cpp