IBatchStream Class Reference

Inheritance diagram for IBatchStream:

Public Member Functions
virtual void	reset (int firstBatch)=0
virtual bool	next ()=0
virtual void	skip (int skipCount)=0
virtual float *	getBatch ()=0
virtual float *	getLabels ()=0
virtual int	getBatchesRead () const =0
virtual int	getBatchSize () const =0
virtual nvinfer1::Dims	getDims () const =0

Member Function Documentation

reset()

virtual void IBatchStream::reset ( int  firstBatch )

pure virtual

Implemented in BatchStream, MNISTBatchStream, and BatchStream.

next()

virtual bool IBatchStream::next ( )

pure virtual

Implemented in BatchStream, MNISTBatchStream, and BatchStream.

skip()

virtual void IBatchStream::skip ( int  skipCount )

pure virtual

Implemented in BatchStream, MNISTBatchStream, and BatchStream.

getBatch()

virtual float* IBatchStream::getBatch ( )

pure virtual

Implemented in BatchStream, MNISTBatchStream, and BatchStream.

getLabels()

virtual float* IBatchStream::getLabels ( )

pure virtual

Implemented in BatchStream, MNISTBatchStream, and BatchStream.

getBatchesRead()

virtual int IBatchStream::getBatchesRead ( ) const

pure virtual

Implemented in BatchStream, MNISTBatchStream, and BatchStream.

getBatchSize()

virtual int IBatchStream::getBatchSize ( ) const

pure virtual

Implemented in BatchStream, MNISTBatchStream, and BatchStream.

getDims()

virtual nvinfer1::Dims IBatchStream::getDims ( ) const

pure virtual

Implemented in BatchStream, MNISTBatchStream, and BatchStream.

---

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

• BatchStream.h