From 83ca5d6afd387caf6abb4767d2f415c16ee02139 Mon Sep 17 00:00:00 2001 From: Yuefeng Zhou Date: Wed, 12 Apr 2017 15:38:38 -0800 Subject: Make "node_name:0" a legit target node name. Change: 153004183 --- tensorflow/core/graph/subgraph.cc | 5 +---- 1 file changed, 1 insertion(+), 4 deletions(-) diff --git a/tensorflow/core/graph/subgraph.cc b/tensorflow/core/graph/subgraph.cc index ff46abd439..91292500e1 100644 --- a/tensorflow/core/graph/subgraph.cc +++ b/tensorflow/core/graph/subgraph.cc @@ -143,10 +143,7 @@ static bool AddNodeToTargets(const string& node_or_tensor_name, return false; } const Node* n = iter->second; - if (n->name() != node_or_tensor_name) { - return false; - } - + CHECK_EQ(n->name(), id.first); targets->insert(n); return true; } -- cgit v1.2.3