Searched refs:_create_chunk_dtensor (Results 1 – 3 of 3) sorted by relevance
6 _create_chunk_dtensor,65 dtensor = _create_chunk_dtensor(tensor, self.rank, device_mesh)
10 _create_chunk_dtensor,148 else _create_chunk_dtensor
92 def _create_chunk_dtensor( function