diff options
-rw-r--r-- | tensorflow/core/graph/subgraph.cc | 5 |
1 files changed, 1 insertions, 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; } |