Searched refs:summed_device_grad_packs (Results 1 – 1 of 1) sorted by relevance
533 def unpack(self, summed_device_grad_packs): argument536 for (summed_device_grad_packs,538 summed_device_grad_packs, self.grouped_grads_and_vars,543 with ops.colocate_with(summed_device_grad_packs[0][0]):545 device_grad_packs = [g for g, _ in summed_device_grad_packs]602 def unpack(self, summed_device_grad_packs): argument604 return cross_device_utils.unpack_small_tensors(summed_device_grad_packs,