Searched defs:ShardMetadata (Results 1 – 4 of 4) sorted by relevance
7 def _check_shard_metadata_pair_overlap(shard1: ShardMetadata, shard2: ShardMetadata):27 shards: List[ShardMetadata], sharded_dims: List[int]58 shards: List[ShardMetadata], dim: int72 def validate_non_overlapping_shards_metadata(shards: List[ShardMetadata]):
47 def _chunk_for_shard(shard_md: ShardMetadata) -> ChunkStorageMetadata:55 sharded_tensor: ShardedTensor, shard_md: ShardMetadata95 fqn: str, sharded_tensor: ShardedTensor, shard_md: ShardMetadata
28 def narrow_tensor(tensor: torch.Tensor, metadata: ShardMetadata) -> torch.Tensor:
10 class ShardMetadata: class