Searched refs:_chunk_tensor (Results 1 – 1 of 1) sorted by relevance
146 def _chunk_tensor( function367 return _chunk_tensor(tensor, rank, world_size, num_devices_per_node, pg)