Ejemplo n.º 1
0
 def unpack(self, summed_device_grad_packs):
   """Reverse the aggregation process."""
   return cross_tower_utils.unpack_small_tensors(summed_device_grad_packs,
                                                 self.packing)
Ejemplo n.º 2
0
 def unpack(self, summed_device_grad_packs):
   """Reverse the aggregation process."""
   return cross_tower_utils.unpack_small_tensors(summed_device_grad_packs,
                                                 self.packing)