Searched refs:tensor_packer (Results 1 – 1 of 1) sorted by relevance
796 tensor_packer = _ConcatAndSplitPacker(num_packs)797 device_grad_packs = tensor_packer.pack(device_grads)799 tensor_packer = None801 return device_grad_packs, tensor_packer804 def _unpack_tensors(reduced, tensor_packer=None): argument806 if tensor_packer:807 return tensor_packer.unpack(reduced)887 device_grad_packs, tensor_packer = _pack_tensors(grouped, self._num_packs)903 reduced = _unpack_tensors(reduced, tensor_packer)