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)
