Logo ROOT   6.18/05
Reference Guide
List of all members | Public Member Functions | Private Types | Private Attributes | List of all members
TMVA::DNN::TTensorDataLoader< Data_t, Architecture_t > Class Template Reference

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_tThe input data type.
Architecture_tThe achitecture class of the underlying architecture.

Definition at line 130 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 ()
 
void CopyTensorInput (HostBuffer_t &buffer, IndexIterator_t begin)
 Copy input tensor into the given host buffer. More...
 
void CopyTensorInput (std::vector< TMatrixT< Double_t > > &tensor, IndexIterator_t sampleIterator)
 
void CopyTensorInput (std::vector< TMatrixT< Double_t > > &tensor, IndexIterator_t sampleIterator)
 
void CopyTensorInput (std::vector< TMatrixT< Real_t > > &tensor, IndexIterator_t sampleIterator)
 
void CopyTensorInput (std::vector< TMatrixT< Real_t > > &tensor, IndexIterator_t sampleIterator)
 
void CopyTensorInput (TCpuBuffer< Double_t > &buffer, IndexIterator_t sampleIterator)
 
void CopyTensorInput (TCpuBuffer< Double_t > &buffer, IndexIterator_t sampleIterator)
 
void CopyTensorInput (TCpuBuffer< Real_t > &buffer, IndexIterator_t sampleIterator)
 
void CopyTensorInput (TCpuBuffer< Real_t > &buffer, IndexIterator_t sampleIterator)
 
void CopyTensorInput (TCudaHostBuffer< double > &buffer, IndexIterator_t sampleIterator)
 
void CopyTensorInput (TCudaHostBuffer< double > &buffer, IndexIterator_t sampleIterator)
 
void CopyTensorInput (TCudaHostBuffer< float > &buffer, IndexIterator_t sampleIterator)
 
void CopyTensorInput (TCudaHostBuffer< float > &buffer, IndexIterator_t sampleIterator)
 
void CopyTensorOutput (HostBuffer_t &buffer, IndexIterator_t begin)
 Copy output matrix into the given host buffer. More...
 
void CopyTensorOutput (TCpuBuffer< Double_t > &buffer, IndexIterator_t sampleIterator)
 
void CopyTensorOutput (TCpuBuffer< Double_t > &buffer, IndexIterator_t sampleIterator)
 
void CopyTensorOutput (TCpuBuffer< Real_t > &buffer, IndexIterator_t sampleIterator)
 
void CopyTensorOutput (TCpuBuffer< Real_t > &buffer, IndexIterator_t sampleIterator)
 
void CopyTensorOutput (TCudaHostBuffer< double > &buffer, IndexIterator_t sampleIterator)
 
void CopyTensorOutput (TCudaHostBuffer< double > &buffer, IndexIterator_t sampleIterator)
 
void CopyTensorOutput (TCudaHostBuffer< float > &buffer, IndexIterator_t sampleIterator)
 
void CopyTensorOutput (TCudaHostBuffer< float > &buffer, IndexIterator_t sampleIterator)
 
void CopyTensorOutput (TMatrixT< Double_t > &matrix, IndexIterator_t sampleIterator)
 
void CopyTensorOutput (TMatrixT< Double_t > &matrix, IndexIterator_t sampleIterator)
 
void CopyTensorOutput (TMatrixT< Real_t > &matrix, IndexIterator_t sampleIterator)
 
void CopyTensorOutput (TMatrixT< Real_t > &matrix, IndexIterator_t sampleIterator)
 
void CopyTensorWeights (HostBuffer_t &buffer, IndexIterator_t begin)
 Copy weight matrix into the given host buffer. More...
 
void CopyTensorWeights (TCpuBuffer< Double_t > &buffer, IndexIterator_t sampleIterator)
 
void CopyTensorWeights (TCpuBuffer< Double_t > &buffer, IndexIterator_t sampleIterator)
 
void CopyTensorWeights (TCpuBuffer< Real_t > &buffer, IndexIterator_t sampleIterator)
 
void CopyTensorWeights (TCpuBuffer< Real_t > &buffer, IndexIterator_t sampleIterator)
 
void CopyTensorWeights (TCudaHostBuffer< double > &buffer, IndexIterator_t sampleIterator)
 
void CopyTensorWeights (TCudaHostBuffer< double > &buffer, IndexIterator_t sampleIterator)
 
void CopyTensorWeights (TCudaHostBuffer< float > &buffer, IndexIterator_t sampleIterator)
 
void CopyTensorWeights (TCudaHostBuffer< float > &buffer, IndexIterator_t sampleIterator)
 
void CopyTensorWeights (TMatrixT< Double_t > &matrix, IndexIterator_t sampleIterator)
 
void CopyTensorWeights (TMatrixT< Double_t > &matrix, IndexIterator_t sampleIterator)
 
void CopyTensorWeights (TMatrixT< Real_t > &matrix, IndexIterator_t sampleIterator)
 
void CopyTensorWeights (TMatrixT< Real_t > &matrix, IndexIterator_t sampleIterator)
 
BatchIterator_t end ()
 
TTensorBatch< Architecture_t > GetTensorBatch ()
 Return the next batch from the training set. More...
 
TTensorDataLoaderoperator= (const TTensorDataLoader &)=default
 
TTensorDataLoaderoperator= (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_tfDeviceBuffers
 The device buffers used to keep the input, output and weight data. More...
 
std::vector< HostBuffer_tfHostBuffers
 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<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() [2/13]

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]

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() [4/13]

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() [5/13]

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() [6/13]

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() [7/13]

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() [8/13]

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() [9/13]

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() [10/13]

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

Definition at line 497 of file CudaBuffers.cxx.

◆ CopyTensorInput() [11/13]

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

Definition at line 546 of file CudaBuffers.cxx.

◆ CopyTensorInput() [12/13]

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

Definition at line 364 of file CudaBuffers.cxx.

◆ CopyTensorInput() [13/13]

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

Definition at line 413 of file CudaBuffers.cxx.

◆ CopyTensorOutput() [1/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() [2/13]

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() [3/13]

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() [4/13]

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() [5/13]

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() [6/13]

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

Definition at line 516 of file CudaBuffers.cxx.

◆ CopyTensorOutput() [7/13]

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

Definition at line 584 of file CudaBuffers.cxx.

◆ CopyTensorOutput() [8/13]

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

Definition at line 383 of file CudaBuffers.cxx.

◆ CopyTensorOutput() [9/13]

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]

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() [11/13]

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() [12/13]

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() [13/13]

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.

◆ CopyTensorWeights() [1/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() [2/13]

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() [3/13]

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() [4/13]

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() [5/13]

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() [6/13]

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

Definition at line 534 of file CudaBuffers.cxx.

◆ CopyTensorWeights() [7/13]

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

Definition at line 619 of file CudaBuffers.cxx.

◆ CopyTensorWeights() [8/13]

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

Definition at line 401 of file CudaBuffers.cxx.

◆ CopyTensorWeights() [9/13]

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]

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() [11/13]

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() [12/13]

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() [13/13]

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.

◆ 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.


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