Searched refs:device_grads (Results 1 – 1 of 1) sorted by relevance
793 def _pack_tensors(device_grads, num_packs=0): argument797 device_grad_packs = tensor_packer.pack(device_grads)800 device_grad_packs = device_grads