Searched refs:_pack_tensors (Results 1 – 1 of 1) sorted by relevance
793 def _pack_tensors(device_grads, num_packs=0): function887 device_grad_packs, tensor_packer = _pack_tensors(grouped, self._num_packs)