Searched refs:state_map_ (Results 1 – 3 of 3) sorted by relevance
287 : state_map_(graph), library_(library), graph_(graph) {} in FunctionalizeCond()341 StateMap* state_map_; member in tensorflow::functionalize_cond::Conditional379 : parent_(parent), state_map_(cond_state_map), predicate_(predicate) {} in Conditional()503 state_map_->ResetCondId(switch_node, state_map_->LookupCondId(src)); in AddSwitchNodeAlongEdge()504 state_map_->ResetAncestorId(switch_node, state_map_->LookupAncestorId(src)); in AddSwitchNodeAlongEdge()521 const auto& id = state_map_->LookupCondId(e->src()); in ExtractBodies()523 : state_map_->FindBranchOf(id, predicate_); in ExtractBodies()530 << state_map_->CondStateToString(m); in ExtractBodies()539 << DebugString(state_map_->LookupCondId(e->src())) << "]."; in ExtractBodies()573 auto dst_id = state_map_->LookupCondId(dst); in ExtractBodies()[all …]
41 return fc_->state_map_.GetCondId(state); in GetUniqueId()45 return fc_->state_map_.CondStateToString(id); in GetString()
257 StateMap state_map_; variable