Searched refs:summed_device_grad_packs (Results 1 – 1 of 1) sorted by relevance
758 def unpack(self, summed_device_grad_packs): argument761 for (summed_device_grad_packs,763 summed_device_grad_packs, self.grouped_grads_and_vars,768 with ops.colocate_with(summed_device_grad_packs[0][0]):770 device_grad_packs = [g for g, _ in summed_device_grad_packs]