Mila
Deep Neural Network Library
Loading...
Searching...
No Matches
Mila::Data::TokenSequenceLoader< TMemoryResource > Member List

This is the complete list of members for Mila::Data::TokenSequenceLoader< TMemoryResource >, including all inherited members.

BaseLoader typedef (defined in Mila::Data::TokenSequenceLoader< TMemoryResource >)Mila::Data::TokenSequenceLoader< TMemoryResource >
batchSize() const noexceptMila::Data::DataLoader< TensorDataType::INT32, TensorDataType::INT32, TMemoryResource >inline
currentBatch() const noexceptMila::Data::DataLoader< TensorDataType::INT32, TensorDataType::INT32, TMemoryResource >inline
DataLoader(int64_t batch_size)Mila::Data::DataLoader< TensorDataType::INT32, TensorDataType::INT32, TMemoryResource >inlineexplicit
DataLoader(const DataLoader &)=deleteMila::Data::DataLoader< TensorDataType::INT32, TensorDataType::INT32, TMemoryResource >
DataLoader(DataLoader &&)=defaultMila::Data::DataLoader< TensorDataType::INT32, TensorDataType::INT32, TMemoryResource >
getDatasetInfo() constMila::Data::DataLoader< TensorDataType::INT32, TensorDataType::INT32, TMemoryResource >inlinevirtual
hasNext() constMila::Data::DataLoader< TensorDataType::INT32, TensorDataType::INT32, TMemoryResource >inlinevirtual
HostType typedef (defined in Mila::Data::TokenSequenceLoader< TMemoryResource >)Mila::Data::TokenSequenceLoader< TMemoryResource >
incrementBatch() noexceptMila::Data::DataLoader< TensorDataType::INT32, TensorDataType::INT32, TMemoryResource >inlineprotected
input_data_typeMila::Data::DataLoader< TensorDataType::INT32, TensorDataType::INT32, TMemoryResource >static
InputDataType typedefMila::Data::DataLoader< TensorDataType::INT32, TensorDataType::INT32, TMemoryResource >
inputs() overrideMila::Data::TokenSequenceLoader< TMemoryResource >inlinevirtual
inputs() const overrideMila::Data::TokenSequenceLoader< TMemoryResource >inlinevirtual
InputTensor typedefMila::Data::DataLoader< TensorDataType::INT32, TensorDataType::INT32, TMemoryResource >
is_mixed_precisionMila::Data::DataLoader< TensorDataType::INT32, TensorDataType::INT32, TMemoryResource >static
MemoryResource typedefMila::Data::DataLoader< TensorDataType::INT32, TensorDataType::INT32, TMemoryResource >
nextBatch() overrideMila::Data::TokenSequenceLoader< TMemoryResource >inlinevirtual
numBatches() const overrideMila::Data::TokenSequenceLoader< TMemoryResource >inlinevirtual
numTokens() const (defined in Mila::Data::TokenSequenceLoader< TMemoryResource >)Mila::Data::TokenSequenceLoader< TMemoryResource >inline
numWindows() const (defined in Mila::Data::TokenSequenceLoader< TMemoryResource >)Mila::Data::TokenSequenceLoader< TMemoryResource >inline
operator=(const TokenSequenceLoader &)=delete (defined in Mila::Data::TokenSequenceLoader< TMemoryResource >)Mila::Data::TokenSequenceLoader< TMemoryResource >
operator=(TokenSequenceLoader &&)=delete (defined in Mila::Data::TokenSequenceLoader< TMemoryResource >)Mila::Data::TokenSequenceLoader< TMemoryResource >
reset() overrideMila::Data::TokenSequenceLoader< TMemoryResource >inlinevirtual
sequenceLength() const (defined in Mila::Data::TokenSequenceLoader< TMemoryResource >)Mila::Data::TokenSequenceLoader< TMemoryResource >inline
setCurrentBatch(int64_t batch_index) noexceptMila::Data::DataLoader< TensorDataType::INT32, TensorDataType::INT32, TMemoryResource >inlineprotected
supportsMixedPrecision() noexceptMila::Data::DataLoader< TensorDataType::INT32, TensorDataType::INT32, TMemoryResource >inlinestatic
target_data_typeMila::Data::DataLoader< TensorDataType::INT32, TensorDataType::INT32, TMemoryResource >static
TargetDataType typedefMila::Data::DataLoader< TensorDataType::INT32, TensorDataType::INT32, TMemoryResource >
targets() overrideMila::Data::TokenSequenceLoader< TMemoryResource >inlinevirtual
targets() const overrideMila::Data::TokenSequenceLoader< TMemoryResource >inlinevirtual
TargetTensor typedefMila::Data::DataLoader< TensorDataType::INT32, TensorDataType::INT32, TMemoryResource >
TensorType typedef (defined in Mila::Data::TokenSequenceLoader< TMemoryResource >)Mila::Data::TokenSequenceLoader< TMemoryResource >
TokenSequenceLoader(const std::filesystem::path &tokens_file, int64_t batch_size, int64_t seq_length, bool is_training, DeviceId device, const TokenSequenceLoaderConfig &config=TokenSequenceLoaderConfig())Mila::Data::TokenSequenceLoader< TMemoryResource >inline
TokenSequenceLoader(const TokenSequenceLoader &)=delete (defined in Mila::Data::TokenSequenceLoader< TMemoryResource >)Mila::Data::TokenSequenceLoader< TMemoryResource >
TokenSequenceLoader(TokenSequenceLoader &&)=delete (defined in Mila::Data::TokenSequenceLoader< TMemoryResource >)Mila::Data::TokenSequenceLoader< TMemoryResource >
uses_pinned_memoryMila::Data::DataLoader< TensorDataType::INT32, TensorDataType::INT32, TMemoryResource >static
usesPinnedMemory() noexceptMila::Data::DataLoader< TensorDataType::INT32, TensorDataType::INT32, TMemoryResource >inlinestatic
validateCurrentBatch() constMila::Data::DataLoader< TensorDataType::INT32, TensorDataType::INT32, TMemoryResource >inlinevirtual
windowSizeTokens() const (defined in Mila::Data::TokenSequenceLoader< TMemoryResource >)Mila::Data::TokenSequenceLoader< TMemoryResource >inline
~DataLoader()=defaultMila::Data::DataLoader< TensorDataType::INT32, TensorDataType::INT32, TMemoryResource >virtual
~TokenSequenceLoader() noexcept (defined in Mila::Data::TokenSequenceLoader< TMemoryResource >)Mila::Data::TokenSequenceLoader< TMemoryResource >inline