diff options
Diffstat (limited to 'tensorflow/core/common_runtime/graph_runner.cc')
| -rw-r--r-- | tensorflow/core/common_runtime/graph_runner.cc | 9 |
1 files changed, 5 insertions, 4 deletions
diff --git a/tensorflow/core/common_runtime/graph_runner.cc b/tensorflow/core/common_runtime/graph_runner.cc index 2ce1e8b483..d0f9e6ed18 100644 --- a/tensorflow/core/common_runtime/graph_runner.cc +++ b/tensorflow/core/common_runtime/graph_runner.cc @@ -123,8 +123,8 @@ Status GraphRunner::Run(Graph* graph, FunctionLibraryRuntime* function_library, for (const auto& in : inputs) { const string& tensor_name = in.first; input_names.emplace_back(tensor_name); - string full_key = Rendezvous::CreateKey("/cpu:0", 1, "/cpu:1", tensor_name, - FrameAndIter(0, 0)); + string full_key = Rendezvous::CreateKey("/device:CPU:0", 1, "/device:CPU:1", + tensor_name, FrameAndIter(0, 0)); Rendezvous::ParsedKey parsed; TF_RETURN_IF_ERROR(Rendezvous::ParseKey(full_key, &parsed)); TF_RETURN_IF_ERROR(rendez->Send(parsed, Rendezvous::Args(), in.second, @@ -175,8 +175,9 @@ Status GraphRunner::Run(Graph* graph, FunctionLibraryRuntime* function_library, outputs->resize(output_names.size()); for (size_t i = 0; i < output_names.size(); ++i) { - const string& output_key = Rendezvous::CreateKey( - "/cpu:0", 1, "/cpu:1", output_names[i], FrameAndIter(0, 0)); + const string& output_key = + Rendezvous::CreateKey("/device:CPU:0", 1, "/device:CPU:1", + output_names[i], FrameAndIter(0, 0)); Rendezvous::ParsedKey parsed; TF_RETURN_IF_ERROR(Rendezvous::ParseKey(output_key, &parsed)); bool is_dead; |