template<typename Data_t, typename Architecture_t>
class TMVA::DNN::TTensorDataLoader< Data_t, Architecture_t >

TTensorDataLoader.

Service class managing the streaming of the training data from the input data type to the accelerator device or the CPU. A TTensorDataLoader object manages a number of host and device buffer pairs that are used in a round-robin manner for the transfer of batches to the device.

Each TTensorDataLoader object has an associated batch size and a number of total samples in the dataset. One epoch is the number of buffers required to transfer the complete training set. Using the begin() and end() member functions allows the user to iterate over the batches in one epoch.

Template Parameters

Data_t	The input data type.
Architecture_t	The achitecture class of the underlying architecture.

Definition at line 82 of file TensorDataLoader.h.

Public Member Functions
	TTensorDataLoader (const Data_t &data, size_t nSamples, size_t batchSize, size_t batchDepth, size_t batchHeight, size_t batchWidth, size_t nOutputFeatures, size_t nStreams=1)
	Constructor. More...

	TTensorDataLoader (const TTensorDataLoader &)=default

	TTensorDataLoader (TTensorDataLoader &&)=default

BatchIterator_t	begin ()

template<>
void	CopyTensorInput (std::vector< TMatrixT< Real_t >> &tensor, IndexIterator_t sampleIterator)

template<>
void	CopyTensorInput (std::vector< TMatrixT< Double_t >> &tensor, IndexIterator_t sampleIterator)

template<>
void	CopyTensorInput (std::vector< TMatrixT< Real_t >> &tensor, IndexIterator_t sampleIterator)

void	CopyTensorInput (HostBuffer_t &buffer, IndexIterator_t begin)
	Copy input tensor into the given host buffer. More...

template<>
void	CopyTensorInput (std::vector< TMatrixT< Double_t >> &tensor, IndexIterator_t sampleIterator)

template<>
void	CopyTensorInput (TCpuBuffer< Real_t > &buffer, IndexIterator_t sampleIterator)

template<>
void	CopyTensorInput (TCpuBuffer< Double_t > &buffer, IndexIterator_t sampleIterator)

template<>
void	CopyTensorInput (TCudaHostBuffer< float > &buffer, IndexIterator_t sampleIterator)

template<>
void	CopyTensorInput (TCudaHostBuffer< float > &buffer, IndexIterator_t sampleIterator)

template<>
void	CopyTensorInput (TCpuBuffer< Double_t > &buffer, IndexIterator_t sampleIterator)

template<>
void	CopyTensorInput (TCudaHostBuffer< double > &buffer, IndexIterator_t sampleIterator)

template<>
void	CopyTensorInput (TCpuBuffer< Real_t > &buffer, IndexIterator_t sampleIterator)

template<>
void	CopyTensorInput (TCudaHostBuffer< double > &buffer, IndexIterator_t sampleIterator)

template<>
void	CopyTensorOutput (TMatrixT< Real_t > &matrix, IndexIterator_t sampleIterator)

template<>
void	CopyTensorOutput (TMatrixT< Double_t > &matrix, IndexIterator_t sampleIterator)

void	CopyTensorOutput (HostBuffer_t &buffer, IndexIterator_t begin)
	Copy output matrix into the given host buffer. More...

template<>
void	CopyTensorOutput (TMatrixT< Real_t > &matrix, IndexIterator_t sampleIterator)

template<>
void	CopyTensorOutput (TMatrixT< Double_t > &matrix, IndexIterator_t sampleIterator)

template<>
void	CopyTensorOutput (TCpuBuffer< Real_t > &buffer, IndexIterator_t sampleIterator)

template<>
void	CopyTensorOutput (TCudaHostBuffer< float > &buffer, IndexIterator_t sampleIterator)

template<>
void	CopyTensorOutput (TCpuBuffer< Double_t > &buffer, IndexIterator_t sampleIterator)

template<>
void	CopyTensorOutput (TCudaHostBuffer< float > &buffer, IndexIterator_t sampleIterator)

template<>
void	CopyTensorOutput (TCpuBuffer< Double_t > &buffer, IndexIterator_t sampleIterator)

template<>
void	CopyTensorOutput (TCudaHostBuffer< double > &buffer, IndexIterator_t sampleIterator)

template<>
void	CopyTensorOutput (TCpuBuffer< Real_t > &buffer, IndexIterator_t sampleIterator)

template<>
void	CopyTensorOutput (TCudaHostBuffer< double > &buffer, IndexIterator_t sampleIterator)

template<>
void	CopyTensorWeights (TMatrixT< Real_t > &matrix, IndexIterator_t sampleIterator)

template<>
void	CopyTensorWeights (TMatrixT< Double_t > &matrix, IndexIterator_t sampleIterator)

void	CopyTensorWeights (HostBuffer_t &buffer, IndexIterator_t begin)
	Copy weight matrix into the given host buffer. More...

template<>
void	CopyTensorWeights (TMatrixT< Real_t > &matrix, IndexIterator_t sampleIterator)

template<>
void	CopyTensorWeights (TMatrixT< Double_t > &matrix, IndexIterator_t sampleIterator)

template<>
void	CopyTensorWeights (TCpuBuffer< Real_t > &buffer, IndexIterator_t sampleIterator)

template<>
void	CopyTensorWeights (TCudaHostBuffer< float > &buffer, IndexIterator_t sampleIterator)

template<>
void	CopyTensorWeights (TCpuBuffer< Double_t > &buffer, IndexIterator_t sampleIterator)

template<>
void	CopyTensorWeights (TCudaHostBuffer< float > &buffer, IndexIterator_t sampleIterator)

template<>
void	CopyTensorWeights (TCpuBuffer< Double_t > &buffer, IndexIterator_t sampleIterator)

template<>
void	CopyTensorWeights (TCudaHostBuffer< double > &buffer, IndexIterator_t sampleIterator)

template<>
void	CopyTensorWeights (TCpuBuffer< Real_t > &buffer, IndexIterator_t sampleIterator)

template<>
void	CopyTensorWeights (TCudaHostBuffer< double > &buffer, IndexIterator_t sampleIterator)

BatchIterator_t	end ()

TTensorBatch< Architecture_t >	GetTensorBatch ()
	Return the next batch from the training set. More...

TTensorDataLoader &	operator= (const TTensorDataLoader &)=default

TTensorDataLoader &	operator= (TTensorDataLoader &&)=default

template<typename RNG >
void	Shuffle (RNG &rng)
	Shuffle the order of the samples in the batch. More...

Private Types
using	BatchIterator_t = TTensorBatchIterator< Data_t, Architecture_t >

using	DeviceBuffer_t = typename Architecture_t::DeviceBuffer_t

using	HostBuffer_t = typename Architecture_t::HostBuffer_t

using	Matrix_t = typename Architecture_t::Matrix_t

Private Attributes
size_t	fBatchDepth
	The number of matrices in the tensor. More...

size_t	fBatchHeight
	The number od rows in each matrix. More...

size_t	fBatchIndex
	The index of the batch when there are multiple batches in parallel. More...

size_t	fBatchSize
	The size of a batch. More...

size_t	fBatchWidth
	The number of columns in each matrix. More...

const Data_t &	fData
	The data that should be loaded in the batches. More...

std::vector< DeviceBuffer_t >	fDeviceBuffers
	The device buffers used to keep the input, output and weight data. More...

std::vector< HostBuffer_t >	fHostBuffers
	The host buffers used to load the input, output and weight data. More...

size_t	fNOutputFeatures
	The number of outputs from the classifier/regressor. More...

size_t	fNSamples
	The total number of samples in the dataset. More...

size_t	fNStreams
	Number of buffer pairs. More...

std::vector< size_t >	fSampleIndices
	Ordering of the samples in the epoch. More...

#include <TMVA/DNN/TensorDataLoader.h>

Member Typedef Documentation

◆ BatchIterator_t

template<typename Data_t, typename Architecture_t>

using TMVA::DNN::TTensorDataLoader< Data_t, Architecture_t >::BatchIterator_t = TTensorBatchIterator<Data_t, Architecture_t>

private

Definition at line 135 of file TensorDataLoader.h.

◆ DeviceBuffer_t

template<typename Data_t, typename Architecture_t>

using TMVA::DNN::TTensorDataLoader< Data_t, Architecture_t >::DeviceBuffer_t = typename Architecture_t::DeviceBuffer_t

private

Definition at line 133 of file TensorDataLoader.h.

◆ HostBuffer_t

template<typename Data_t, typename Architecture_t>

using TMVA::DNN::TTensorDataLoader< Data_t, Architecture_t >::HostBuffer_t = typename Architecture_t::HostBuffer_t

private

Definition at line 132 of file TensorDataLoader.h.

◆ Matrix_t

template<typename Data_t, typename Architecture_t>

using TMVA::DNN::TTensorDataLoader< Data_t, Architecture_t >::Matrix_t = typename Architecture_t::Matrix_t

private

Definition at line 134 of file TensorDataLoader.h.

Constructor & Destructor Documentation

◆ TTensorDataLoader() [1/3]

template<typename Data_t , typename Architecture_t >

TMVA::DNN::TTensorDataLoader< Data_t, Architecture_t >::TTensorDataLoader	(	const Data_t &	data,
		size_t	nSamples,
		size_t	batchSize,
		size_t	batchDepth,
		size_t	batchHeight,
		size_t	batchWidth,
		size_t	nOutputFeatures,
		size_t	nStreams = `1`
	)

Constructor.

Definition at line 203 of file TensorDataLoader.h.

◆ TTensorDataLoader() [2/3]

template<typename Data_t, typename Architecture_t>

TMVA::DNN::TTensorDataLoader< Data_t, Architecture_t >::TTensorDataLoader ( const TTensorDataLoader< Data_t, Architecture_t > & )

default

◆ TTensorDataLoader() [3/3]

template<typename Data_t, typename Architecture_t>

TMVA::DNN::TTensorDataLoader< Data_t, Architecture_t >::TTensorDataLoader ( TTensorDataLoader< Data_t, Architecture_t > && )

default

Member Function Documentation

◆ begin()

template<typename Data_t, typename Architecture_t>

BatchIterator_t TMVA::DNN::TTensorDataLoader< Data_t, Architecture_t >::begin ( )

inline

Definition at line 173 of file TensorDataLoader.h.

◆ CopyTensorInput() [1/13]

template<>

void TMVA::DNN::TTensorDataLoader< TensorInput, TReference< Real_t > >::CopyTensorInput	(	std::vector< TMatrixT< Real_t >> &	tensor,
		IndexIterator_t	sampleIterator
	)

Definition at line 40 of file TensorDataLoader.cxx.

◆ CopyTensorInput() [2/13]

template<>

void TMVA::DNN::TTensorDataLoader< TensorInput, TReference< Double_t > >::CopyTensorInput	(	std::vector< TMatrixT< Double_t >> &	tensor,
		IndexIterator_t	sampleIterator
	)

Definition at line 100 of file TensorDataLoader.cxx.

◆ CopyTensorInput() [3/13]

template<>

void TMVA::DNN::TTensorDataLoader< TMVAInput_t, TReference< Real_t > >::CopyTensorInput	(	std::vector< TMatrixT< Real_t >> &	tensor,
		IndexIterator_t	sampleIterator
	)

Definition at line 160 of file TensorDataLoader.cxx.

◆ CopyTensorInput() [4/13]

template<typename Data_t, typename Architecture_t>

void TMVA::DNN::TTensorDataLoader< Data_t, Architecture_t >::CopyTensorInput	(	HostBuffer_t &	buffer,
		IndexIterator_t	begin
	)

Copy input tensor into the given host buffer.

Function to be specialized by the architecture-specific backend.

◆ CopyTensorInput() [5/13]

template<>

void TMVA::DNN::TTensorDataLoader< TMVAInput_t, TReference< Double_t > >::CopyTensorInput	(	std::vector< TMatrixT< Double_t >> &	tensor,
		IndexIterator_t	sampleIterator
	)

Definition at line 232 of file TensorDataLoader.cxx.

◆ CopyTensorInput() [6/13]

template<>

void TMVA::DNN::TTensorDataLoader< TensorInput, TCpu< Real_t > >::CopyTensorInput	(	TCpuBuffer< Real_t > &	buffer,
		IndexIterator_t	sampleIterator
	)

Definition at line 299 of file CpuBuffer.cxx.

◆ CopyTensorInput() [7/13]

template<>

void TMVA::DNN::TTensorDataLoader< TensorInput, TCpu< Double_t > >::CopyTensorInput	(	TCpuBuffer< Double_t > &	buffer,
		IndexIterator_t	sampleIterator
	)

Definition at line 361 of file CpuBuffer.cxx.

◆ CopyTensorInput() [8/13]

template<>

void TMVA::DNN::TTensorDataLoader< TensorInput, TCuda< float > >::CopyTensorInput	(	TCudaHostBuffer< float > &	buffer,
		IndexIterator_t	sampleIterator
	)

Definition at line 364 of file CudaBuffers.cxx.

◆ CopyTensorInput() [9/13]

template<>

void TMVA::DNN::TTensorDataLoader< TMVAInput_t, TCuda< float > >::CopyTensorInput	(	TCudaHostBuffer< float > &	buffer,
		IndexIterator_t	sampleIterator
	)

Definition at line 413 of file CudaBuffers.cxx.

◆ CopyTensorInput() [10/13]

template<>

void TMVA::DNN::TTensorDataLoader< TMVAInput_t, TCpu< Double_t > >::CopyTensorInput	(	TCpuBuffer< Double_t > &	buffer,
		IndexIterator_t	sampleIterator
	)

re-implement specialization for Double_t

Definition at line 424 of file CpuBuffer.cxx.

◆ CopyTensorInput() [11/13]

template<>

void TMVA::DNN::TTensorDataLoader< TensorInput, TCuda< double > >::CopyTensorInput	(	TCudaHostBuffer< double > &	buffer,
		IndexIterator_t	sampleIterator
	)

Definition at line 497 of file CudaBuffers.cxx.

◆ CopyTensorInput() [12/13]

template<>

void TMVA::DNN::TTensorDataLoader< TMVAInput_t, TCpu< Real_t > >::CopyTensorInput	(	TCpuBuffer< Real_t > &	buffer,
		IndexIterator_t	sampleIterator
	)

re-implement specialization for Real_t

Definition at line 510 of file CpuBuffer.cxx.

◆ CopyTensorInput() [13/13]

template<>

void TMVA::DNN::TTensorDataLoader< TMVAInput_t, TCuda< double > >::CopyTensorInput	(	TCudaHostBuffer< double > &	buffer,
		IndexIterator_t	sampleIterator
	)

Definition at line 546 of file CudaBuffers.cxx.

◆ CopyTensorOutput() [1/13]

template<>

void TMVA::DNN::TTensorDataLoader< TensorInput, TReference< Real_t > >::CopyTensorOutput	(	TMatrixT< Real_t > &	matrix,
		IndexIterator_t	sampleIterator
	)

Definition at line 68 of file TensorDataLoader.cxx.

◆ CopyTensorOutput() [2/13]

template<>

void TMVA::DNN::TTensorDataLoader< TensorInput, TReference< Double_t > >::CopyTensorOutput	(	TMatrixT< Double_t > &	matrix,
		IndexIterator_t	sampleIterator
	)

Definition at line 128 of file TensorDataLoader.cxx.

◆ CopyTensorOutput() [3/13]

template<typename Data_t, typename Architecture_t>

void TMVA::DNN::TTensorDataLoader< Data_t, Architecture_t >::CopyTensorOutput	(	HostBuffer_t &	buffer,
		IndexIterator_t	begin
	)

Copy output matrix into the given host buffer.

Function to be specialized by the architecture-spcific backend.

◆ CopyTensorOutput() [4/13]

template<>

void TMVA::DNN::TTensorDataLoader< TMVAInput_t, TReference< Real_t > >::CopyTensorOutput	(	TMatrixT< Real_t > &	matrix,
		IndexIterator_t	sampleIterator
	)

Definition at line 191 of file TensorDataLoader.cxx.

◆ CopyTensorOutput() [5/13]

template<>

void TMVA::DNN::TTensorDataLoader< TMVAInput_t, TReference< Double_t > >::CopyTensorOutput	(	TMatrixT< Double_t > &	matrix,
		IndexIterator_t	sampleIterator
	)

Definition at line 263 of file TensorDataLoader.cxx.

◆ CopyTensorOutput() [6/13]

template<>

void TMVA::DNN::TTensorDataLoader< TensorInput, TCpu< Real_t > >::CopyTensorOutput	(	TCpuBuffer< Real_t > &	buffer,
		IndexIterator_t	sampleIterator
	)

Definition at line 329 of file CpuBuffer.cxx.

◆ CopyTensorOutput() [7/13]

template<>

void TMVA::DNN::TTensorDataLoader< TensorInput, TCuda< float > >::CopyTensorOutput	(	TCudaHostBuffer< float > &	buffer,
		IndexIterator_t	sampleIterator
	)

Definition at line 383 of file CudaBuffers.cxx.

◆ CopyTensorOutput() [8/13]

template<>

void TMVA::DNN::TTensorDataLoader< TensorInput, TCpu< Double_t > >::CopyTensorOutput	(	TCpuBuffer< Double_t > &	buffer,
		IndexIterator_t	sampleIterator
	)

Definition at line 391 of file CpuBuffer.cxx.

◆ CopyTensorOutput() [9/13]

template<>

void TMVA::DNN::TTensorDataLoader< TMVAInput_t, TCuda< float > >::CopyTensorOutput	(	TCudaHostBuffer< float > &	buffer,
		IndexIterator_t	sampleIterator
	)

Definition at line 450 of file CudaBuffers.cxx.

◆ CopyTensorOutput() [10/13]

template<>

void TMVA::DNN::TTensorDataLoader< TMVAInput_t, TCpu< Double_t > >::CopyTensorOutput	(	TCpuBuffer< Double_t > &	buffer,
		IndexIterator_t	sampleIterator
	)

Definition at line 462 of file CpuBuffer.cxx.

◆ CopyTensorOutput() [11/13]

template<>

void TMVA::DNN::TTensorDataLoader< TensorInput, TCuda< double > >::CopyTensorOutput	(	TCudaHostBuffer< double > &	buffer,
		IndexIterator_t	sampleIterator
	)

Definition at line 516 of file CudaBuffers.cxx.

◆ CopyTensorOutput() [12/13]

template<>

void TMVA::DNN::TTensorDataLoader< TMVAInput_t, TCpu< Real_t > >::CopyTensorOutput	(	TCpuBuffer< Real_t > &	buffer,
		IndexIterator_t	sampleIterator
	)

Definition at line 548 of file CpuBuffer.cxx.

◆ CopyTensorOutput() [13/13]

template<>

void TMVA::DNN::TTensorDataLoader< TMVAInput_t, TCuda< double > >::CopyTensorOutput	(	TCudaHostBuffer< double > &	buffer,
		IndexIterator_t	sampleIterator
	)

Definition at line 584 of file CudaBuffers.cxx.

◆ CopyTensorWeights() [1/13]

template<>

void TMVA::DNN::TTensorDataLoader< TensorInput, TReference< Real_t > >::CopyTensorWeights	(	TMatrixT< Real_t > &	matrix,
		IndexIterator_t	sampleIterator
	)

Definition at line 86 of file TensorDataLoader.cxx.

◆ CopyTensorWeights() [2/13]

template<>

void TMVA::DNN::TTensorDataLoader< TensorInput, TReference< Double_t > >::CopyTensorWeights	(	TMatrixT< Double_t > &	matrix,
		IndexIterator_t	sampleIterator
	)

Definition at line 146 of file TensorDataLoader.cxx.

◆ CopyTensorWeights() [3/13]

template<typename Data_t, typename Architecture_t>

void TMVA::DNN::TTensorDataLoader< Data_t, Architecture_t >::CopyTensorWeights	(	HostBuffer_t &	buffer,
		IndexIterator_t	begin
	)

Copy weight matrix into the given host buffer.

Function to be specialized by the architecture-spcific backend.

◆ CopyTensorWeights() [4/13]

template<>

void TMVA::DNN::TTensorDataLoader< TMVAInput_t, TReference< Real_t > >::CopyTensorWeights	(	TMatrixT< Real_t > &	matrix,
		IndexIterator_t	sampleIterator
	)

Definition at line 220 of file TensorDataLoader.cxx.

◆ CopyTensorWeights() [5/13]

template<>

void TMVA::DNN::TTensorDataLoader< TMVAInput_t, TReference< Double_t > >::CopyTensorWeights	(	TMatrixT< Double_t > &	matrix,
		IndexIterator_t	sampleIterator
	)

Definition at line 292 of file TensorDataLoader.cxx.

◆ CopyTensorWeights() [6/13]

template<>

void TMVA::DNN::TTensorDataLoader< TensorInput, TCpu< Real_t > >::CopyTensorWeights	(	TCpuBuffer< Real_t > &	buffer,
		IndexIterator_t	sampleIterator
	)

Definition at line 347 of file CpuBuffer.cxx.

◆ CopyTensorWeights() [7/13]

template<>

void TMVA::DNN::TTensorDataLoader< TensorInput, TCuda< float > >::CopyTensorWeights	(	TCudaHostBuffer< float > &	buffer,
		IndexIterator_t	sampleIterator
	)

Definition at line 401 of file CudaBuffers.cxx.

◆ CopyTensorWeights() [8/13]

template<>

void TMVA::DNN::TTensorDataLoader< TensorInput, TCpu< Double_t > >::CopyTensorWeights	(	TCpuBuffer< Double_t > &	buffer,
		IndexIterator_t	sampleIterator
	)

Definition at line 409 of file CpuBuffer.cxx.

◆ CopyTensorWeights() [9/13]

template<>

void TMVA::DNN::TTensorDataLoader< TMVAInput_t, TCuda< float > >::CopyTensorWeights	(	TCudaHostBuffer< float > &	buffer,
		IndexIterator_t	sampleIterator
	)

Definition at line 485 of file CudaBuffers.cxx.

◆ CopyTensorWeights() [10/13]

template<>

void TMVA::DNN::TTensorDataLoader< TMVAInput_t, TCpu< Double_t > >::CopyTensorWeights	(	TCpuBuffer< Double_t > &	buffer,
		IndexIterator_t	sampleIterator
	)

Definition at line 497 of file CpuBuffer.cxx.

◆ CopyTensorWeights() [11/13]

template<>

void TMVA::DNN::TTensorDataLoader< TensorInput, TCuda< double > >::CopyTensorWeights	(	TCudaHostBuffer< double > &	buffer,
		IndexIterator_t	sampleIterator
	)

Definition at line 534 of file CudaBuffers.cxx.

◆ CopyTensorWeights() [12/13]

template<>

void TMVA::DNN::TTensorDataLoader< TMVAInput_t, TCpu< Real_t > >::CopyTensorWeights	(	TCpuBuffer< Real_t > &	buffer,
		IndexIterator_t	sampleIterator
	)

Definition at line 583 of file CpuBuffer.cxx.

◆ CopyTensorWeights() [13/13]

template<>

void TMVA::DNN::TTensorDataLoader< TMVAInput_t, TCuda< double > >::CopyTensorWeights	(	TCudaHostBuffer< double > &	buffer,
		IndexIterator_t	sampleIterator
	)

Definition at line 619 of file CudaBuffers.cxx.

◆ end()

template<typename Data_t, typename Architecture_t>

BatchIterator_t TMVA::DNN::TTensorDataLoader< Data_t, Architecture_t >::end ( )

inline

Definition at line 174 of file TensorDataLoader.h.

◆ GetTensorBatch()

template<typename Data_t , typename Architecture_t >

TTensorBatch< Architecture_t > TMVA::DNN::TTensorDataLoader< Data_t, Architecture_t >::GetTensorBatch ( )

Return the next batch from the training set.

The TTensorDataLoader object keeps an internal counter that cycles over the batches in the training set.

Definition at line 227 of file TensorDataLoader.h.

◆ operator=() [1/2]

template<typename Data_t, typename Architecture_t>

TTensorDataLoader& TMVA::DNN::TTensorDataLoader< Data_t, Architecture_t >::operator= ( const TTensorDataLoader< Data_t, Architecture_t > & )

default

◆ operator=() [2/2]

template<typename Data_t, typename Architecture_t>

TTensorDataLoader& TMVA::DNN::TTensorDataLoader< Data_t, Architecture_t >::operator= ( TTensorDataLoader< Data_t, Architecture_t > && )

default

◆ Shuffle()

template<typename Data_t , typename Architecture_t >

template<typename RNG >

void TMVA::DNN::TTensorDataLoader< Data_t, Architecture_t >::Shuffle ( RNG & rng )

Shuffle the order of the samples in the batch.

The shuffling is indirect, i.e. only the indices are shuffled. No input data is moved by this routine.

Definition at line 276 of file TensorDataLoader.h.

Member Data Documentation

◆ fBatchDepth

template<typename Data_t, typename Architecture_t>

size_t TMVA::DNN::TTensorDataLoader< Data_t, Architecture_t >::fBatchDepth

private

The number of matrices in the tensor.

Definition at line 141 of file TensorDataLoader.h.

◆ fBatchHeight

template<typename Data_t, typename Architecture_t>

size_t TMVA::DNN::TTensorDataLoader< Data_t, Architecture_t >::fBatchHeight

private

The number od rows in each matrix.

Definition at line 142 of file TensorDataLoader.h.

◆ fBatchIndex

template<typename Data_t, typename Architecture_t>

size_t TMVA::DNN::TTensorDataLoader< Data_t, Architecture_t >::fBatchIndex

private

The index of the batch when there are multiple batches in parallel.

Definition at line 145 of file TensorDataLoader.h.

◆ fBatchSize

template<typename Data_t, typename Architecture_t>

size_t TMVA::DNN::TTensorDataLoader< Data_t, Architecture_t >::fBatchSize

private

The size of a batch.

Definition at line 140 of file TensorDataLoader.h.

◆ fBatchWidth

template<typename Data_t, typename Architecture_t>

size_t TMVA::DNN::TTensorDataLoader< Data_t, Architecture_t >::fBatchWidth

private

The number of columns in each matrix.

Definition at line 143 of file TensorDataLoader.h.

◆ fData

template<typename Data_t, typename Architecture_t>

const Data_t& TMVA::DNN::TTensorDataLoader< Data_t, Architecture_t >::fData

private

The data that should be loaded in the batches.

Definition at line 137 of file TensorDataLoader.h.

◆ fDeviceBuffers

template<typename Data_t, typename Architecture_t>

std::vector<DeviceBuffer_t> TMVA::DNN::TTensorDataLoader< Data_t, Architecture_t >::fDeviceBuffers

private

The device buffers used to keep the input, output and weight data.

Definition at line 148 of file TensorDataLoader.h.

◆ fHostBuffers

template<typename Data_t, typename Architecture_t>

std::vector<HostBuffer_t> TMVA::DNN::TTensorDataLoader< Data_t, Architecture_t >::fHostBuffers

private

The host buffers used to load the input, output and weight data.

Definition at line 149 of file TensorDataLoader.h.

◆ fNOutputFeatures

template<typename Data_t, typename Architecture_t>

size_t TMVA::DNN::TTensorDataLoader< Data_t, Architecture_t >::fNOutputFeatures

private

The number of outputs from the classifier/regressor.

Definition at line 144 of file TensorDataLoader.h.

◆ fNSamples

template<typename Data_t, typename Architecture_t>

size_t TMVA::DNN::TTensorDataLoader< Data_t, Architecture_t >::fNSamples

private

The total number of samples in the dataset.

Definition at line 139 of file TensorDataLoader.h.

◆ fNStreams

template<typename Data_t, typename Architecture_t>

size_t TMVA::DNN::TTensorDataLoader< Data_t, Architecture_t >::fNStreams

private

Number of buffer pairs.

Definition at line 147 of file TensorDataLoader.h.

◆ fSampleIndices

template<typename Data_t, typename Architecture_t>

std::vector<size_t> TMVA::DNN::TTensorDataLoader< Data_t, Architecture_t >::fSampleIndices

private

Ordering of the samples in the epoch.

Definition at line 151 of file TensorDataLoader.h.

Libraries for TMVA::DNN::TTensorDataLoader< Data_t, Architecture_t >:

[legend]

template<typename Data_t, typename Architecture_t> class TMVA::DNN::TTensorDataLoader< Data_t, Architecture_t >

Public Member Functions

Private Types

Private Attributes

Member Typedef Documentation

◆ BatchIterator_t

◆ DeviceBuffer_t

◆ HostBuffer_t

◆ Matrix_t

Constructor & Destructor Documentation

◆ TTensorDataLoader() [1/3]

◆ TTensorDataLoader() [2/3]

◆ TTensorDataLoader() [3/3]

Member Function Documentation

◆ begin()

◆ CopyTensorInput() [1/13]

◆ CopyTensorInput() [2/13]

◆ CopyTensorInput() [3/13]

◆ CopyTensorInput() [4/13]

◆ CopyTensorInput() [5/13]

◆ CopyTensorInput() [6/13]

◆ CopyTensorInput() [7/13]

◆ CopyTensorInput() [8/13]

◆ CopyTensorInput() [9/13]

◆ CopyTensorInput() [10/13]

◆ CopyTensorInput() [11/13]

◆ CopyTensorInput() [12/13]

◆ CopyTensorInput() [13/13]

◆ CopyTensorOutput() [1/13]

◆ CopyTensorOutput() [2/13]

◆ CopyTensorOutput() [3/13]

◆ CopyTensorOutput() [4/13]

◆ CopyTensorOutput() [5/13]

◆ CopyTensorOutput() [6/13]

◆ CopyTensorOutput() [7/13]

◆ CopyTensorOutput() [8/13]

◆ CopyTensorOutput() [9/13]

◆ CopyTensorOutput() [10/13]

◆ CopyTensorOutput() [11/13]

◆ CopyTensorOutput() [12/13]

◆ CopyTensorOutput() [13/13]

◆ CopyTensorWeights() [1/13]

◆ CopyTensorWeights() [2/13]

◆ CopyTensorWeights() [3/13]

◆ CopyTensorWeights() [4/13]

◆ CopyTensorWeights() [5/13]

◆ CopyTensorWeights() [6/13]

◆ CopyTensorWeights() [7/13]

◆ CopyTensorWeights() [8/13]

◆ CopyTensorWeights() [9/13]

◆ CopyTensorWeights() [10/13]

◆ CopyTensorWeights() [11/13]

◆ CopyTensorWeights() [12/13]

◆ CopyTensorWeights() [13/13]

◆ end()

◆ GetTensorBatch()

◆ operator=() [1/2]

◆ operator=() [2/2]

◆ Shuffle()

Member Data Documentation

◆ fBatchDepth

◆ fBatchHeight

◆ fBatchIndex

◆ fBatchSize

◆ fBatchWidth

◆ fData

◆ fDeviceBuffers

◆ fHostBuffers

◆ fNOutputFeatures

◆ fNSamples

◆ fNStreams

◆ fSampleIndices

template<typename Data_t, typename Architecture_t>
class TMVA::DNN::TTensorDataLoader< Data_t, Architecture_t >