Searched refs:_create_chunk_from_dtensor (Results 1 – 2 of 2) sorted by relevance
240 def _create_chunk_from_dtensor(tensor: DTensor) -> ChunkStorageMetadata: function258 local_chunks = [_create_chunk_from_dtensor(tensor)]
604 _create_chunk_from_dtensor,610 return [_create_chunk_from_dtensor(self)]