Searched refs:grouped_grads_and_vars (Results 1 – 1 of 1) sorted by relevance
481 def pack(self, grouped_grads_and_vars): argument483 self.grouped_grads_and_vars = grouped_grads_and_vars488 for device_grads_and_vars in grouped_grads_and_vars:538 summed_device_grad_packs, self.grouped_grads_and_vars,592 def pack(self, grouped_grads_and_vars): argument597 grouped_grads_and_vars,