Home
last modified time | relevance | path

Searched defs:ChunkStorageMetadata (Results 1 – 4 of 4) sorted by relevance

/aosp_15_r20/external/pytorch/torch/distributed/checkpoint/
H A Dresharding.py11 shard1: ChunkStorageMetadata, shard2: ChunkStorageMetadata
29 saved_shard: ChunkStorageMetadata, current_shard: ChunkStorageMetadata
H A Ddefault_planner.py471 def _check_box_overlap(box0: ChunkStorageMetadata, box1: ChunkStorageMetadata) -> bool:
488 outer_box_size: torch.Size, inner_box: ChunkStorageMetadata
H A Dmetadata.py23 class ChunkStorageMetadata: class
H A Dplanner_helpers.py154 local_chunks: List[ChunkStorageMetadata],