Searched refs:cond_arg_nodes_ (Results 1 – 1 of 1) sorted by relevance
368 CondArgNodes cond_arg_nodes_; member in tensorflow::functionalize_cond::Conditional417 input_index[key] = cond_arg_nodes_.size(); in BuildArgumentNodes()418 cond_arg_nodes_.emplace_back(key.first, key.second); in BuildArgumentNodes()420 cond_arg_nodes_.at(input_index.at(key)).switches.push_back(switch_node); in BuildArgumentNodes()422 VLOG(5) << "CondArg nodes created: " << DebugString(cond_arg_nodes_); in BuildArgumentNodes()425 for (CondArgNode& cond_arg_node : cond_arg_nodes_) { in BuildArgumentNodes()752 for (auto& kv : cond_arg_nodes_) { in BuildIfNode()828 for (auto& arg : cond_arg_nodes_) { in AddInputEdges()