diff options
| author |  Yuefeng Zhou <yuefengz@google.com> | 2017-04-12 15:38:38 -0800 |
|---|---|---|
| committer |  TensorFlower Gardener <gardener@tensorflow.org> | 2017-04-12 16:50:19 -0700 |
| commit | 83ca5d6afd387caf6abb4767d2f415c16ee02139 (patch) | |
| tree | 1d29504c0c85dcb5854a25787a9b696c605938c8 | |
| parent | a5a8558feb9417359e30a991ab5e01cf17194473 (diff) | |
Make "node_name:0" a legit target node name.
Change: 153004183
| -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; } |