| allocateBuffers() | Mila::Data::TokenSequenceLoader< TMemoryResource > | inlineprivate |
| back_buffer_ready_ | Mila::Data::TokenSequenceLoader< TMemoryResource > | private |
| back_input_tensor_ | Mila::Data::TokenSequenceLoader< TMemoryResource > | private |
| back_target_tensor_ | Mila::Data::TokenSequenceLoader< TMemoryResource > | private |
| BaseLoader typedef | Mila::Data::TokenSequenceLoader< TMemoryResource > | |
| batch_size_ | Mila::Data::DataLoader< TensorDataType::INT32, TensorDataType::INT32, TMemoryResource > | private |
| batches_per_window_ | Mila::Data::TokenSequenceLoader< TMemoryResource > | private |
| batchSize() const noexcept | Mila::Data::DataLoader< TensorDataType::INT32, TensorDataType::INT32, TMemoryResource > | inline |
| cleanupBuffers() noexcept | Mila::Data::TokenSequenceLoader< TMemoryResource > | inlineprivate |
| config_ | Mila::Data::TokenSequenceLoader< TMemoryResource > | private |
| current_batch_ | Mila::Data::DataLoader< TensorDataType::INT32, TensorDataType::INT32, TMemoryResource > | private |
| current_batch_in_window_ | Mila::Data::TokenSequenceLoader< TMemoryResource > | private |
| current_window_idx_ | Mila::Data::TokenSequenceLoader< TMemoryResource > | private |
| currentBatch() const noexcept | Mila::Data::DataLoader< TensorDataType::INT32, TensorDataType::INT32, TMemoryResource > | inline |
| cv_consumer_ | Mila::Data::TokenSequenceLoader< TMemoryResource > | private |
| cv_producer_ | Mila::Data::TokenSequenceLoader< TMemoryResource > | private |
| DataLoader(int64_t batch_size) | Mila::Data::DataLoader< TensorDataType::INT32, TensorDataType::INT32, TMemoryResource > | inlineexplicit |
| DataLoader(const DataLoader &)=delete | Mila::Data::DataLoader< TensorDataType::INT32, TensorDataType::INT32, TMemoryResource > | |
| DataLoader(DataLoader &&)=default | Mila::Data::DataLoader< TensorDataType::INT32, TensorDataType::INT32, TMemoryResource > | |
| device_ | Mila::Data::TokenSequenceLoader< TMemoryResource > | private |
| file_size_ | Mila::Data::TokenSequenceLoader< TMemoryResource > | private |
| fillBatch(const TokenId *window_buffer, size_t batch_idx, HostType *input_dest, HostType *target_dest) | Mila::Data::TokenSequenceLoader< TMemoryResource > | inlineprivate |
| front_buffer_ready_ | Mila::Data::TokenSequenceLoader< TMemoryResource > | private |
| front_input_tensor_ | Mila::Data::TokenSequenceLoader< TMemoryResource > | private |
| front_target_tensor_ | Mila::Data::TokenSequenceLoader< TMemoryResource > | private |
| getDatasetInfo() const | Mila::Data::DataLoader< TensorDataType::INT32, TensorDataType::INT32, TMemoryResource > | inlinevirtual |
| hasNext() const | Mila::Data::DataLoader< TensorDataType::INT32, TensorDataType::INT32, TMemoryResource > | inlinevirtual |
| HostType typedef | Mila::Data::TokenSequenceLoader< TMemoryResource > | |
| incrementBatch() noexcept | Mila::Data::DataLoader< TensorDataType::INT32, TensorDataType::INT32, TMemoryResource > | inlineprotected |
| initializeDataset() | Mila::Data::TokenSequenceLoader< TMemoryResource > | inlineprivate |
| input_data_type | Mila::Data::DataLoader< TensorDataType::INT32, TensorDataType::INT32, TMemoryResource > | static |
| InputDataType typedef | Mila::Data::DataLoader< TensorDataType::INT32, TensorDataType::INT32, TMemoryResource > | |
| inputs() override | Mila::Data::TokenSequenceLoader< TMemoryResource > | inlinevirtual |
| inputs() const override | Mila::Data::TokenSequenceLoader< TMemoryResource > | inlinevirtual |
| InputTensor typedef | Mila::Data::DataLoader< TensorDataType::INT32, TensorDataType::INT32, TMemoryResource > | |
| is_mixed_precision | Mila::Data::DataLoader< TensorDataType::INT32, TensorDataType::INT32, TMemoryResource > | static |
| is_training_ | Mila::Data::TokenSequenceLoader< TMemoryResource > | private |
| loadWindowFromFile(std::ifstream &file, TokenId *buffer, size_t window_idx) | Mila::Data::TokenSequenceLoader< TMemoryResource > | inlineprivate |
| MemoryResource typedef | Mila::Data::DataLoader< TensorDataType::INT32, TensorDataType::INT32, TMemoryResource > | |
| mutex_ | Mila::Data::TokenSequenceLoader< TMemoryResource > | private |
| nextBatch() override | Mila::Data::TokenSequenceLoader< TMemoryResource > | inlinevirtual |
| num_batches_ | Mila::Data::TokenSequenceLoader< TMemoryResource > | private |
| num_tokens_ | Mila::Data::TokenSequenceLoader< TMemoryResource > | private |
| num_windows_ | Mila::Data::TokenSequenceLoader< TMemoryResource > | private |
| numBatches() const override | Mila::Data::TokenSequenceLoader< TMemoryResource > | inlinevirtual |
| numTokens() const | Mila::Data::TokenSequenceLoader< TMemoryResource > | inline |
| numWindows() const | Mila::Data::TokenSequenceLoader< TMemoryResource > | inline |
| operator=(const TokenSequenceLoader &)=delete | Mila::Data::TokenSequenceLoader< TMemoryResource > | |
| operator=(TokenSequenceLoader &&)=delete | Mila::Data::TokenSequenceLoader< TMemoryResource > | |
| Mila::Data::DataLoader< TensorDataType::INT32, TensorDataType::INT32, TMemoryResource >::operator=(const DataLoader &)=delete | Mila::Data::DataLoader< TensorDataType::INT32, TensorDataType::INT32, TMemoryResource > | |
| Mila::Data::DataLoader< TensorDataType::INT32, TensorDataType::INT32, TMemoryResource >::operator=(DataLoader &&)=default | Mila::Data::DataLoader< TensorDataType::INT32, TensorDataType::INT32, TMemoryResource > | |
| prepareSequenceIndices() | Mila::Data::TokenSequenceLoader< TMemoryResource > | inlineprivate |
| producer_exception_ | Mila::Data::TokenSequenceLoader< TMemoryResource > | private |
| producer_thread_ | Mila::Data::TokenSequenceLoader< TMemoryResource > | private |
| producerThreadFunc() noexcept | Mila::Data::TokenSequenceLoader< TMemoryResource > | inlineprivate |
| reset() override | Mila::Data::TokenSequenceLoader< TMemoryResource > | inlinevirtual |
| seq_length_ | Mila::Data::TokenSequenceLoader< TMemoryResource > | private |
| sequence_indices_ | Mila::Data::TokenSequenceLoader< TMemoryResource > | private |
| sequenceLength() const | Mila::Data::TokenSequenceLoader< TMemoryResource > | inline |
| sequences_per_window_ | Mila::Data::TokenSequenceLoader< TMemoryResource > | private |
| setCurrentBatch(int64_t batch_index) noexcept | Mila::Data::DataLoader< TensorDataType::INT32, TensorDataType::INT32, TMemoryResource > | inlineprotected |
| shuffleSequenceIndices() | Mila::Data::TokenSequenceLoader< TMemoryResource > | inlineprivate |
| stop_ | Mila::Data::TokenSequenceLoader< TMemoryResource > | private |
| supportsMixedPrecision() noexcept | Mila::Data::DataLoader< TensorDataType::INT32, TensorDataType::INT32, TMemoryResource > | inlinestatic |
| swapBuffers() noexcept | Mila::Data::TokenSequenceLoader< TMemoryResource > | inlineprivate |
| target_data_type | Mila::Data::DataLoader< TensorDataType::INT32, TensorDataType::INT32, TMemoryResource > | static |
| TargetDataType typedef | Mila::Data::DataLoader< TensorDataType::INT32, TensorDataType::INT32, TMemoryResource > | |
| targets() override | Mila::Data::TokenSequenceLoader< TMemoryResource > | inlinevirtual |
| targets() const override | Mila::Data::TokenSequenceLoader< TMemoryResource > | inlinevirtual |
| TargetTensor typedef | Mila::Data::DataLoader< TensorDataType::INT32, TensorDataType::INT32, TMemoryResource > | |
| TensorType typedef | Mila::Data::TokenSequenceLoader< TMemoryResource > | |
| tokens_file_path_ | Mila::Data::TokenSequenceLoader< TMemoryResource > | private |
| 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 | Mila::Data::TokenSequenceLoader< TMemoryResource > | |
| TokenSequenceLoader(TokenSequenceLoader &&)=delete | Mila::Data::TokenSequenceLoader< TMemoryResource > | |
| uses_pinned_memory | Mila::Data::DataLoader< TensorDataType::INT32, TensorDataType::INT32, TMemoryResource > | static |
| usesPinnedMemory() noexcept | Mila::Data::DataLoader< TensorDataType::INT32, TensorDataType::INT32, TMemoryResource > | inlinestatic |
| validateCurrentBatch() const | Mila::Data::DataLoader< TensorDataType::INT32, TensorDataType::INT32, TMemoryResource > | inlinevirtual |
| validateDeviceId(DeviceId device) | Mila::Data::TokenSequenceLoader< TMemoryResource > | inlineprivatestatic |
| window_size_tokens_ | Mila::Data::TokenSequenceLoader< TMemoryResource > | private |
| windowSizeTokens() const | Mila::Data::TokenSequenceLoader< TMemoryResource > | inline |
| ~DataLoader()=default | Mila::Data::DataLoader< TensorDataType::INT32, TensorDataType::INT32, TMemoryResource > | virtual |
| ~TokenSequenceLoader() noexcept | Mila::Data::TokenSequenceLoader< TMemoryResource > | inline |
1.15.0