def unpack(self, summed_device_grad_packs): """Reverse the aggregation process.""" return cross_tower_utils.unpack_small_tensors(summed_device_grad_packs, self.packing)
def unpack(self, summed_device_grad_packs): """Reverse the aggregation process.""" return cross_tower_utils.unpack_small_tensors(summed_device_grad_packs, self.packing)