Args Struct Reference

Collaboration diagram for Args:

Public Attributes
bool	help {false}
std::string	engine {}
std::vector< int >	batchSize
int	sequenceLength {128}
int	iterations {200}
int	warmUpRuns {10}
int	randomSeed {12345}
bool	enableGraph {false}
int32_t	embeddingVecSize {EMBEDDING_VEC_SIZE}
int32_t	numUsers {NUM_USERS}
int32_t	topKMovies {TOPK_MOVIES}
int32_t	numMoviesPerUser {NUM_INDICES}
int32_t	nbProcesses {THREADS}
std::string	weightFile {DEFAULT_WEIGHT_FILE}
std::string	ratingInputFile {RATING_INPUT_FILE}
std::string	uffFile {UFF_MODEL_FILE}
std::string	engineFile {ENGINE_FILE}
bool	enableFP16 {false}
bool	enableInt8 {false}
bool	enableVerbose {false}
int	useDLACore {-1}
std::map< int32_t, std::vector< int32_t > >	userToItemsMap
std::map< int32_t, std::vector< std::pair< int32_t, float > > >	userToExpectedItemProbMap
int32_t	device {DEVICE}
std::vector< OutputArgs >	pargsVec
std::atomic< int32_t >	failCount

Member Data Documentation

help

bool Args::help {false}

engine

std::string Args::engine {}

batchSize

std::vector<int> Args::batchSize

sequenceLength

int Args::sequenceLength {128}

iterations

int Args::iterations {200}

warmUpRuns

int Args::warmUpRuns {10}

randomSeed

int Args::randomSeed {12345}

enableGraph

bool Args::enableGraph {false}

embeddingVecSize

int32_t Args::embeddingVecSize {EMBEDDING_VEC_SIZE}

numUsers

int32_t Args::numUsers {NUM_USERS}

topKMovies

int32_t Args::topKMovies {TOPK_MOVIES}

numMoviesPerUser

int32_t Args::numMoviesPerUser {NUM_INDICES}

nbProcesses

int32_t Args::nbProcesses {THREADS}

weightFile

std::string Args::weightFile {DEFAULT_WEIGHT_FILE}

ratingInputFile

std::string Args::ratingInputFile {RATING_INPUT_FILE}

uffFile

std::string Args::uffFile {UFF_MODEL_FILE}

engineFile

std::string Args::engineFile {ENGINE_FILE}

enableFP16

bool Args::enableFP16 {false}

enableInt8

bool Args::enableInt8 {false}

enableVerbose

bool Args::enableVerbose {false}

useDLACore

int Args::useDLACore {-1}

userToItemsMap

std::map<int32_t, std::vector<int32_t> > Args::userToItemsMap

userToExpectedItemProbMap

std::map<int32_t, std::vector<std::pair<int32_t, float> > > Args::userToExpectedItemProbMap

device

int32_t Args::device {DEVICE}

pargsVec

std::vector<OutputArgs> Args::pargsVec

failCount

std::atomic<int32_t> Args::failCount

---

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

• demo/BERT/infer_c/common.h
• sampleMovieLensMPS.cpp