experiments/STT-tensorflow
1
0
Fork 0
Code Issues Pull Requests Projects Releases Wiki Activity

Update tensorflow/compiler/tf2xla/functionalize_cond.cc

Browse Source
This commit is contained in:
Mihai Maruseac 2020-07-28 22:53:13 +00:00 committed by GitHub
parent 1701747714
commit 3aaf9e186b
No known key found for this signature in database
GPG Key ID: 4AEE18F83AFDEB23
1 changed files with 2 additions and 2 deletions

4
tensorflow/compiler/tf2xla/functionalize_cond.cc
View File

@ -224,8 +224,8 @@ string DebugString(const CondArgNodes& nodes) {
}
StateMap::CondId StateMap::LookupCondId(const Node* node) const {
const int64 node_to_condid_map_size = node_to_condid_map_.size();
if (node->id() < node_to_condid_map_size)
const int64 map_size = node_to_condid_map_.size();
if (node->id() < map_size)
return node_to_condid_map_[node->id()];
return added_node_condid_mapping_.at(node->id());
}