Home
last modified time | relevance | path

Searched refs:LookupAncestorId (Results 1 – 2 of 2) sorted by relevance

/external/tensorflow/tensorflow/compiler/tf2xla/
Dfunctionalize_cond.h124 AncestorId LookupAncestorId(const Node* node) const;
Dfunctionalize_cond.cc246 StateMap::AncestorId StateMap::LookupAncestorId(const Node* node) const { in LookupAncestorId() function in tensorflow::functionalize_cond::StateMap
277 if (auto id = LookupAncestorId(node)) { in AncestorStateToString()
520 state_map_->ResetAncestorId(switch_node, state_map_->LookupAncestorId(src)); in AddSwitchNodeAlongEdge()
973 state_map_.ResetAncestorId(id, state_map_.LookupAncestorId(if_node)); in AddIdentityNode()
993 state_map_.ResetAncestorId(ret, state_map_.LookupAncestorId(replacee)); in AddIfNode()
1343 auto other_id = state_map_.LookupAncestorId(src); in DetermineAncestorState()
1557 cond_id, state_map_.LookupAncestorId(merge), predicate->second); in FunctionalizeInternal()