Home
last modified time | relevance | path

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

/external/tensorflow/tensorflow/compiler/tf2xla/
Dfunctionalize_cond.cc365 std::array<std::vector<Node*>, 2> node_maps_; member in tensorflow::functionalize_cond::Conditional
442 Node* dst_copy = node_maps_[branch_index][e->dst()->id()]; in BuildArgumentNodes()
466 for (auto e : node_maps_[static_cast<int>(branch)][m->id()]->in_edges()) { in BuildArgumentNodes()
557 node_maps_[branch_index].resize(graph->num_node_ids(), nullptr); in ExtractBodies()
558 auto& node_map = node_maps_[branch_index]; in ExtractBodies()
694 auto& node_map = node_maps_[branch_index]; in ExtractBodies()
706 auto& node_map = node_maps_[branch_index]; in ExtractBodies()