Index _ | A | B | C | D | E | F | G | I | L | M | N | P | R | S | T | U | W | Z _ __getitem__() (grain.experimental.FlatMapMapDataset method) (grain.experimental.WindowShuffleMapDataset method) (grain.experimental.ZipMapDataset method) (grain.MapDataset method) (grain.samplers.Sampler method) (grain.sources.RandomAccessDataSource method) __init__() (grain.DataLoader method) (grain.DataLoaderIterator method) (grain.DatasetIterator method) (grain.experimental.BestFitPackIterDataset method) (grain.experimental.CacheIterDataset method) (grain.experimental.ConcatThenSplitIterDataset method) (grain.experimental.ElasticIterator method) (grain.experimental.FirstFitPackIterDataset method) (grain.experimental.FlatMapIterDataset method) (grain.experimental.FlatMapMapDataset method) (grain.experimental.InterleaveIterDataset method) (grain.experimental.LimitIterDataset method) (grain.experimental.ParquetIterDataset method) (grain.experimental.RebatchIterDataset method) (grain.experimental.RepeatIterDataset method) (grain.experimental.TFRecordIterDataset method) (grain.experimental.ThreadPrefetchDatasetIterator method) (grain.experimental.ThreadPrefetchIterDataset method) (grain.experimental.WindowShuffleIterDataset method) (grain.experimental.WindowShuffleMapDataset method) (grain.experimental.WithOptionsIterDataset method) (grain.experimental.ZipIterDataset method) (grain.experimental.ZipMapDataset method) (grain.IterDataset method) (grain.MapDataset method) (grain.samplers.IndexSampler method) (grain.samplers.Sampler method) (grain.samplers.SequentialSampler method) (grain.sources.ArrayRecordDataSource method) (grain.sources.RangeDataSource method) (grain.sources.SharedMemoryDataSource method) __iter__() (grain.DataLoader method) (grain.DataLoaderIterator method) (grain.DatasetIterator method) (grain.experimental.BestFitPackIterDataset method) (grain.experimental.CacheIterDataset method) (grain.experimental.ConcatThenSplitIterDataset method) (grain.experimental.FirstFitPackIterDataset method) (grain.experimental.FlatMapIterDataset method) (grain.experimental.InterleaveIterDataset method) (grain.experimental.LimitIterDataset method) (grain.experimental.ParquetIterDataset method) (grain.experimental.RebatchIterDataset method) (grain.experimental.RepeatIterDataset method) (grain.experimental.TFRecordIterDataset method) (grain.experimental.ThreadPrefetchIterDataset method) (grain.experimental.WindowShuffleIterDataset method) (grain.experimental.WithOptionsIterDataset method) (grain.experimental.ZipIterDataset method) (grain.IterDataset method) (grain.MapDataset method) __len__() (grain.MapDataset method) (grain.sources.RandomAccessDataSource method) __next__() (grain.DataLoaderIterator method) (grain.experimental.ElasticIterator method) (grain.experimental.ThreadPrefetchDatasetIterator method) A acquire_rng() (grain.experimental.RngPool method) apply() (grain.IterDataset method) (grain.MapDataset method) apply_transformations() (in module grain.experimental) ArrayRecordDataSource (class in grain.sources) assert_equal_output_after_checkpoint() (in module grain.experimental) B Batch (class in grain.transforms) batch() (grain.IterDataset method) (grain.MapDataset method) batch_and_pad() (in module grain.experimental) BestFitPackIterDataset (class in grain.experimental) BOSHandling (class in grain.experimental) C CacheIterDataset (class in grain.experimental) CheckpointHandler (class in grain.checkpoint) CheckpointRestore (class in grain.checkpoint) CheckpointSave (class in grain.checkpoint) close() (grain.DatasetIterator method) concatenate() (_src.python.dataset.dataset.MapDatasetMeta method) ConcatThenSplitIterDataset (class in grain.experimental) count (grain.experimental.LimitIterDataset attribute) D data (grain.Record attribute) DataLoader (class in grain) DataLoaderIterator (class in grain) DATASET_INDEX (in module grain.constants) DatasetIterator (class in grain) DatasetOptions (class in grain.experimental) DatasetSelectionMap (class in grain.transforms) device_put() (in module grain.experimental) drop_remainder (grain.sharding.ShardOptions attribute) E ElasticIterator (class in grain.experimental) enable_async_del() (grain.multiprocessing.SharedMemoryArray class method) enable_profiling (grain.multiprocessing.MultiprocessingOptions attribute) EPOCH (in module grain.constants) execution_tracking_mode (grain.experimental.DatasetOptions attribute) ExecutionTrackingMode (class in grain.experimental) F Filter (class in grain.transforms) filter() (grain.IterDataset method) (grain.MapDataset method) filter_raise_threshold_ratio (grain.experimental.DatasetOptions attribute) filter_warn_threshold_ratio (grain.experimental.DatasetOptions attribute) FirstFitPackIterDataset (class in grain.experimental) FlatMapIterDataset (class in grain.experimental) FlatMapMapDataset (class in grain.experimental) FlatMapTransform (in module grain.experimental) G get_element_spec() (in module grain.experimental) get_next_index() (in module grain.experimental) get_state() (grain.DataLoaderIterator method) (grain.DatasetIterator method) grain._src.python.data_loader module grain._src.python.dataset.dataset module grain.checkpoint module grain.experimental module grain.multiprocessing module grain.samplers module grain.sharding module grain.sources module grain.transforms module I index (grain.RecordMetadata attribute) INDEX (in module grain.constants) index_shuffle() (in module grain.experimental) IndexSampler (class in grain.samplers) InterleaveIterDataset (class in grain.experimental) IterDataset (class in grain) IterDatasetMeta (class in _src.python.dataset.dataset) L LimitIterDataset (class in grain.experimental) load() (grain.DataLoaderIterator method) (grain.DatasetIterator method) (in module grain) M Map (class in grain.transforms) map() (grain.IterDataset method) (grain.MapDataset method) map_with_index() (grain.IterDataset method) (grain.MapDataset method) MapDataset (class in grain) MapDatasetMeta (class in _src.python.dataset.dataset) MapWithIndex (class in grain.transforms) merge() (grain.experimental.DatasetOptions method) META_FEATURES (in module grain.constants) metadata (grain.Record attribute) min_shm_size (grain.experimental.DatasetOptions attribute) mix() (_src.python.dataset.dataset.IterDatasetMeta method) (_src.python.dataset.dataset.MapDatasetMeta method) module grain._src.python.data_loader grain._src.python.dataset.dataset grain.checkpoint grain.experimental grain.multiprocessing grain.samplers grain.sharding grain.sources grain.transforms mp_prefetch() (grain.IterDataset method) multiprocessing_options (grain.DataLoader property) MultiprocessingOptions (class in grain.multiprocessing) multithread_prefetch() (in module grain.experimental) N NoSharding (class in grain.sharding) num_epochs (grain.samplers.IndexSampler attribute) num_records (grain.samplers.IndexSampler attribute) num_threads (grain.ReadOptions attribute) num_workers (grain.multiprocessing.MultiprocessingOptions attribute) P parent (grain.experimental.LimitIterDataset attribute) (grain.experimental.ThreadPrefetchIterDataset attribute) parents (grain.IterDataset property) (grain.MapDataset property) ParquetIterDataset (class in grain.experimental) per_worker_buffer_size (grain.multiprocessing.MultiprocessingOptions attribute) PerformanceConfig (class in grain.experimental) pick_performance_config() (in module grain.experimental) pipe() (grain.IterDataset method) (grain.MapDataset method) prefetch() (grain.IterDataset method) prefetch_buffer_size (grain.experimental.ThreadPrefetchIterDataset attribute) (grain.ReadOptions attribute) R random_map() (grain.IterDataset method) (grain.MapDataset method) RandomAccessDataSource (class in grain.sources) RandomMap (class in grain.transforms) range() (_src.python.dataset.dataset.MapDatasetMeta method) RangeDataSource (class in grain.sources) ReadOptions (class in grain) RebatchIterDataset (class in grain.experimental) Record (class in grain) RECORD (in module grain.constants) record_key (grain.RecordMetadata attribute) RECORD_KEY (in module grain.constants) RecordMetadata (class in grain) remove_record_key() (grain.RecordMetadata method) repeat() (grain.MapDataset method) RepeatIterDataset (class in grain.experimental) restore() (grain.checkpoint.CheckpointHandler method) rng (grain.RecordMetadata attribute) RngPool (class in grain.experimental) S Sampler (class in grain.samplers) save() (grain.checkpoint.CheckpointHandler method) (grain.DataLoaderIterator method) (grain.DatasetIterator method) seed (grain.samplers.IndexSampler attribute) SEED (in module grain.constants) seed() (grain.IterDataset method) (grain.MapDataset method) select_from_datasets() (_src.python.dataset.dataset.MapDatasetMeta method) SequentialSampler (class in grain.samplers) set_next_index() (in module grain.experimental) set_state() (grain.DataLoaderIterator method) (grain.DatasetIterator method) shard_count (grain.sharding.ShardOptions attribute) shard_index (grain.sharding.ShardOptions attribute) shard_options (grain.samplers.IndexSampler attribute) ShardByJaxProcess (class in grain.sharding) ShardOptions (class in grain.sharding) SharedMemoryArray (class in grain.multiprocessing) SharedMemoryDataSource (class in grain.sources) shuffle (grain.samplers.IndexSampler attribute) shuffle() (grain.MapDataset method) slice() (grain.MapDataset method) source() (_src.python.dataset.dataset.MapDatasetMeta method) start_prefetch() (grain.DatasetIterator method) T TFRecordIterDataset (class in grain.experimental) ThreadPrefetchDatasetIterator (class in grain.experimental) ThreadPrefetchIterDataset (class in grain.experimental) to_iter_dataset() (grain.MapDataset method) Transformation (in module grain.transforms) Transformations (in module grain.transforms) U unlink_on_del() (grain.multiprocessing.SharedMemoryArray method) W WindowShuffleIterDataset (class in grain.experimental) WindowShuffleMapDataset (class in grain.experimental) WithOptionsIterDataset (class in grain.experimental) Z ZipIterDataset (class in grain.experimental) ZipMapDataset (class in grain.experimental)