MNISTBatchStream Class Reference

Inheritance diagram for MNISTBatchStream:

Collaboration diagram for MNISTBatchStream:

Public Member Functions
	MNISTBatchStream (int batchSize, int maxBatches, const std::string &dataFile, const std::string &labelsFile, const std::vector< std::string > &directories)
void	reset (int firstBatch) override
bool	next () override
void	skip (int skipCount) override
float *	getBatch () override
float *	getLabels () override
int	getBatchesRead () const override
int	getBatchSize () const override
nvinfer1::Dims	getDims () const override

Private Member Functions
void	readDataFile (const std::string &dataFilePath)
void	readLabelsFile (const std::string &labelsFilePath)

Private Attributes
int	mBatchSize {0}
int	mBatchCount {0}
	The batch that will be read on the next invocation of next() More...
int	mMaxBatches {0}
Dims	mDims {}
std::vector< float >	mData {}
std::vector< float >	mLabels {}

Constructor & Destructor Documentation

MNISTBatchStream()

MNISTBatchStream::MNISTBatchStream ( int  batchSize, int  maxBatches, const std::string &  dataFile, const std::string &  labelsFile, const std::vector< std::string > &  directories  )

inline

Member Function Documentation

reset()

void MNISTBatchStream::reset ( int  firstBatch )

inlineoverridevirtual

Implements IBatchStream.

next()

bool MNISTBatchStream::next ( )

inlineoverridevirtual

Implements IBatchStream.

Here is the caller graph for this function:

skip()

void MNISTBatchStream::skip ( int  skipCount )

inlineoverridevirtual

Implements IBatchStream.

Here is the caller graph for this function:

getBatch()

float* MNISTBatchStream::getBatch ( )

inlineoverridevirtual

Implements IBatchStream.

Here is the call graph for this function:

Here is the caller graph for this function:

getLabels()

float* MNISTBatchStream::getLabels ( )

inlineoverridevirtual

Implements IBatchStream.

Here is the caller graph for this function:

getBatchesRead()

int MNISTBatchStream::getBatchesRead ( ) const

inlineoverridevirtual

Implements IBatchStream.

Here is the caller graph for this function:

getBatchSize()

int MNISTBatchStream::getBatchSize ( ) const

inlineoverridevirtual

Implements IBatchStream.

getDims()

nvinfer1::Dims MNISTBatchStream::getDims ( ) const

inlineoverridevirtual

Implements IBatchStream.

readDataFile()

void MNISTBatchStream::readDataFile ( const std::string &  dataFilePath )

inlineprivate

Here is the call graph for this function:

readLabelsFile()

void MNISTBatchStream::readLabelsFile ( const std::string &  labelsFilePath )

inlineprivate

Here is the call graph for this function:

Member Data Documentation

mBatchSize

int MNISTBatchStream::mBatchSize {0}

private

mBatchCount

int MNISTBatchStream::mBatchCount {0}

private

The batch that will be read on the next invocation of next()

mMaxBatches

int MNISTBatchStream::mMaxBatches {0}

private

mDims

Dims MNISTBatchStream::mDims {}

private

mData

std::vector<float> MNISTBatchStream::mData {}

private

mLabels

std::vector<float> MNISTBatchStream::mLabels {}

private

---

The documentation for this class was generated from the following file:

• BatchStream.h