Searched refs:cond_arg_nodes_ (Results 1 – 1 of 1) sorted by relevance
381 CondArgNodes cond_arg_nodes_; member in tensorflow::functionalize_cond::Conditional433 input_index[key] = cond_arg_nodes_.size(); in BuildArgumentNodes()434 cond_arg_nodes_.emplace_back(key.first, key.second); in BuildArgumentNodes()436 cond_arg_nodes_.at(input_index.at(key)).switches.push_back(switch_node); in BuildArgumentNodes()438 VLOG(5) << "CondArg nodes created: " << DebugString(cond_arg_nodes_); in BuildArgumentNodes()441 for (CondArgNode& cond_arg_node : cond_arg_nodes_) { in BuildArgumentNodes()778 for (auto& kv : cond_arg_nodes_) { in BuildIfNode()869 for (auto& arg : cond_arg_nodes_) { in AddInputEdges()