Searched defs:shards_metadata (Results 1 – 3 of 3) sorted by relevance
90 shards_metadata: List[ShardMetadata] = field(default_factory=list) variable in ShardedTensorMetadata
114 def check_tensor(shards_metadata, tensor_dims) -> None: argument
186 def _infer_sharding_spec_from_shards_metadata(shards_metadata): argument