diff options
author | 2017-04-11 11:12:15 -0800 | |
---|---|---|
committer | 2017-04-11 12:23:53 -0700 | |
commit | a11e669c5a4c855b0b507da97378bc7e03a08f86 (patch) | |
tree | 341ba99ba25ed30c2148dbf91d6a41e86d56ad82 /tensorflow/core/kernels/remote_fused_graph_execute_utils.cc | |
parent | 740053be5054027bc66c7995c045d40b07479943 (diff) |
Support two data types as inputs in RemoteFusedGraphExecuteOp
Change: 152843285
Diffstat (limited to 'tensorflow/core/kernels/remote_fused_graph_execute_utils.cc')
-rw-r--r-- | tensorflow/core/kernels/remote_fused_graph_execute_utils.cc | 3 |
1 files changed, 2 insertions, 1 deletions
diff --git a/tensorflow/core/kernels/remote_fused_graph_execute_utils.cc b/tensorflow/core/kernels/remote_fused_graph_execute_utils.cc index a6eb06004a..b885c5ce09 100644 --- a/tensorflow/core/kernels/remote_fused_graph_execute_utils.cc +++ b/tensorflow/core/kernels/remote_fused_graph_execute_utils.cc @@ -187,7 +187,8 @@ RemoteFusedGraphExecuteUtils::GetExecutorBuildRegistry() { const std::vector<std::pair<string, Tensor>>& input_tensor_vector, const string& node_name) { for (const std::pair<string, Tensor>& pair : input_tensor_vector) { - if (node_name == pair.first) { + const TensorId tid = ParseTensorName(pair.first); + if (node_name == tid.first.ToString()) { return true; } } |