Searched refs:_pack_tensors (Results 1 – 1 of 1) sorted by relevance
608 def _pack_tensors(device_grads, function722 device_grad_packs, tensor_packer = _pack_tensors(908 device_grad_packs, tensor_packer = _pack_tensors(