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

Apply cl/347541110 to the soon-to-be refactored custom device support

Browse Source 
Just want to make sure I don't clobber the logic when the refactor gets checked in.

PiperOrigin-RevId: 347726465
Change-Id: Id0849534c8f0e4021aebcacd25181cc77f6fbdc9
This commit is contained in:
Allen Lavoie 2020-12-15 17:33:27 -08:00 committed by TensorFlower Gardener
parent 1ce28a3f8a
commit 74b755bff1
1 changed files with 4 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

5
tensorflow/core/common_runtime/eager/custom_device.cc
View File

@ -58,7 +58,10 @@ int CustomDeviceTensorHandle::DeviceId(Status* status) const {
AbstractTensorInterface* CustomDeviceTensorHandle::Resolve(Status* status) { AbstractTensorInterface* CustomDeviceTensorHandle::Resolve(Status* status) {
core::RefCountPtr<ImmediateExecutionTensorHandle> copied_off( core::RefCountPtr<ImmediateExecutionTensorHandle> copied_off(
context_->CopyTensorHandleToDevice( context_->CopyTensorHandleToDevice(
this, "/job:localhost/replica:0/task:0/device:CPU:0", status)); this,
DeviceNameUtils::ParsedNameToString(context_->HostCPUParsedName())
.c_str(),
status));
if (!status->ok()) { if (!status->ok()) {
return nullptr; return nullptr;
} }