experiments
/
STT-tensorflow
1
0
Fork 0
Code Issues Pull Requests Projects Releases Wiki Activity

Fix for mac build break. 

PiperOrigin-RevId: 304747923
Change-Id: If6fdb5bbea59869f9fc0f9a5ec06eff042b94cbe
This commit is contained in:
A. Unique TensorFlower 2020-04-03 21:45:32 -07:00 committed by TensorFlower Gardener
parent e530f61b9e
commit 802e00d37e
1 changed files with 3 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

4
tensorflow/compiler/xla/python/local_client.cc
View File

@ -776,7 +776,9 @@ PyLocalBuffer::CopyToDeviceHelper(
return copy_event_or.status();
}
return std::pair(std::move(py_buffer), copy_event_or.ConsumeValueOrDie());
return std::pair<std::unique_ptr<PyLocalBuffer>,
std::shared_ptr<BufferDefinitionEvent>>(
std::move(py_buffer), copy_event_or.ConsumeValueOrDie());
}
StatusOr<std::unique_ptr<PyLocalBuffer>> PyLocalBuffer::CopyToDevice(