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

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

Collaboration diagram for magmadnn::dataloader::LinearLoader< T >:

Public Member Functions
	LinearLoader (Tensor< T > *x, Tensor< T > *y, unsigned int batch_size)
virtual void	next (Tensor< T > *x_batch, Tensor< T > *y_batch)
virtual void	reset ()
Public Member Functions inherited from magmadnn::dataloader::DataLoader< T >
	DataLoader (Tensor< T > *x, Tensor< T > *y, unsigned int batch_size)
virtual unsigned int	get_batch_size () const
virtual void	set_batch_size (unsigned int size)
virtual unsigned int	get_num_batches () const

Additional Inherited Members
Protected Attributes inherited from magmadnn::dataloader::DataLoader< T >
Tensor< T > *	x
Tensor< T > *	y
unsigned int	batch_size
unsigned int	sample_size_x
unsigned int	sample_size_y
unsigned int	num_batches

Member Function Documentation

next()

template<typename T >

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

virtual

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

Parameters

x_batch
y_batch

Implements magmadnn::dataloader::DataLoader< T >.

Here is the call graph for this function:

reset()

template<typename T >

void magmadnn::dataloader::LinearLoader< T >::reset ( )

virtual

Resets the dataloader for the next epoch

Implements magmadnn::dataloader::DataLoader< T >.

---

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

• include/dataloader/linear/linearloader.h
• src/dataloader/linear/linearloader.cpp