magmadnn::dataloader::DataLoader< T > Class Template Reference

Inheritance diagram for magmadnn::dataloader::DataLoader< T >:

Public Member Functions
	DataLoader (Tensor< T > *x, Tensor< T > *y, unsigned int batch_size)
virtual void	next (Tensor< T > *x_batch, Tensor< T > *y_batch)=0
virtual void	reset ()=0
virtual unsigned int	get_batch_size () const
virtual void	set_batch_size (unsigned int size)
virtual unsigned int	get_num_batches () const

Protected Attributes
Tensor< T > *	x
Tensor< T > *	y
unsigned int	batch_size
unsigned int	sample_size_x
unsigned int	sample_size_y
unsigned int	num_batches

Constructor & Destructor Documentation

DataLoader()

template<typename T >

magmadnn::dataloader::DataLoader< T >::DataLoader ( Tensor< T > *  x, Tensor< T > *  y, unsigned int  batch_size  )

inline

Constructs a DataLoader object with given data and parameters

Parameters

x	input features
y	output labels
batch_size

Here is the call graph for this function:

Member Function Documentation

next()

template<typename T >

virtual void magmadnn::dataloader::DataLoader< T >::next ( Tensor< T > *  x_batch, Tensor< T > *  y_batch  )

pure virtual

Copies the next batch of inputs and outputs to x_batch and y_batch

Parameters

x_batch
y_batch

Implemented in magmadnn::dataloader::LinearLoader< T >.

Here is the caller graph for this function:

reset()

template<typename T >

virtual void magmadnn::dataloader::DataLoader< T >::reset ( )

pure virtual

Resets the dataloader for the next epoch

Implemented in magmadnn::dataloader::LinearLoader< T >.

Here is the caller graph for this function:

---

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

• include/dataloader/dataloader.h