Index _ | A | B | C | D | E | F | G | I | L | M | N | P | R | S | T | W | Z _ __getitem__() (grain.MapDataset method) __init__() (grain.checkpoint.CheckpointHandler method) (grain.DataLoader method) (grain.DataLoaderIterator method) (grain.DatasetIterator method) (grain.experimental.BOSHandling method) (grain.experimental.ConcatThenSplitIterDataset method) (grain.experimental.DatasetOptions method) (grain.experimental.ElasticIterator method) (grain.experimental.ExecutionTrackingMode method) (grain.experimental.FirstFitPackIterDataset method) (grain.experimental.FlatMapIterDataset method) (grain.experimental.FlatMapMapDataset method) (grain.experimental.FlatMapTransform method) (grain.experimental.InterleaveIterDataset method) (grain.experimental.LimitIterDataset method) (grain.experimental.PackAndBatchOperation method) (grain.experimental.ParquetIterDataset method) (grain.experimental.RngPool 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.multiprocessing.MultiprocessingOptions method) (grain.multiprocessing.SharedMemoryArray method) (grain.Record method) (grain.RecordMetadata method) (grain.samplers.IndexSampler method) (grain.samplers.Sampler method) (grain.samplers.SequentialSampler method) (grain.sharding.NoSharding method) (grain.sharding.ShardByJaxProcess method) (grain.sharding.ShardOptions method) (grain.sources.ArrayRecordDataSource method) (grain.sources.RandomAccessDataSource method) (grain.sources.RangeDataSource method) (grain.sources.SharedMemoryDataSource method) (grain.transforms.Batch method) (grain.transforms.DatasetSelectionMap method) (grain.transforms.Filter method) (grain.transforms.MapWithIndex method) __iter__() (grain.DatasetIterator method) (grain.IterDataset method) (grain.MapDataset method) __len__() (grain.MapDataset method) A 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) batch_size (grain.experimental.PackAndBatchOperation attribute) BOSHandling (class in grain.experimental) C CheckpointHandler (class in grain.checkpoint) CheckpointRestore (in module grain.checkpoint) CheckpointSave (in module grain.checkpoint) concatenate() (_src.python.dataset.dataset.MapDatasetMeta method) ConcatThenSplitIterDataset (class in grain.experimental) count (grain.experimental.LimitIterDataset attribute) D 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_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 (class in grain.experimental) G get_state() (grain.DatasetIterator method) grain._src.python.dataset.dataset module grain.checkpoint module grain.constants module grain.experimental module grain.multiprocessing module grain.samplers module grain.sharding module grain.sources module grain.transforms module I INDEX (in module grain.constants) IndexSampler (class in grain.samplers) InterleaveIterDataset (class in grain.experimental) IterDataset (class in grain) IterDatasetMeta (class in _src.python.dataset.dataset) L length_struct (grain.experimental.PackAndBatchOperation attribute) LimitIterDataset (class in grain.experimental) load() (grain.DatasetIterator method) (in module grain) M Map (in module 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) META_FEATURES (in module grain.constants) min_shm_size (grain.experimental.DatasetOptions attribute) mix() (_src.python.dataset.dataset.IterDatasetMeta method) (_src.python.dataset.dataset.MapDatasetMeta method) module grain._src.python.dataset.dataset grain.checkpoint grain.constants grain.experimental grain.multiprocessing grain.samplers grain.sharding grain.sources grain.transforms mp_prefetch() (grain.IterDataset method) MultiprocessingOptions (class in grain.multiprocessing) N NoSharding (class in grain.sharding) num_workers (grain.multiprocessing.MultiprocessingOptions attribute) P PackAndBatchOperation (class in grain.experimental) 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) pipe() (grain.IterDataset method) (grain.MapDataset method) prefetch() (grain.IterDataset method) prefetch_buffer_size (grain.experimental.ThreadPrefetchIterDataset attribute) R random_map() (grain.IterDataset method) (grain.MapDataset method) RandomAccessDataSource (class in grain.sources) RandomMap (in module grain.transforms) range() (_src.python.dataset.dataset.MapDatasetMeta method) RangeDataSource (class in grain.sources) Record (class in grain) RECORD (in module grain.constants) RECORD_KEY (in module grain.constants) RecordMetadata (class in grain) repeat() (grain.MapDataset method) RngPool (class in grain.experimental) S Sampler (class in grain.samplers) save() (grain.DatasetIterator method) 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_state() (grain.DatasetIterator method) shard_count (grain.sharding.ShardOptions attribute) shard_index (grain.sharding.ShardOptions attribute) ShardByJaxProcess (class in grain.sharding) ShardOptions (class in grain.sharding) SharedMemoryArray (class in grain.multiprocessing) SharedMemoryDataSource (class in grain.sources) shuffle() (grain.MapDataset method) slice() (grain.MapDataset method) source() (_src.python.dataset.dataset.MapDatasetMeta method) start_prefetch() (grain.DatasetIterator method) T 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) 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)