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.cc378 std::array<std::vector<Node*>, 2> node_maps_; member in tensorflow::functionalize_cond::Conditional
458 Node* dst_copy = node_maps_[branch_index][e->dst()->id()]; in BuildArgumentNodes()
482 for (auto e : node_maps_[static_cast<int>(branch)][m->id()]->in_edges()) { in BuildArgumentNodes()
573 node_maps_[branch_index].resize(graph->num_node_ids(), nullptr); in ExtractBodies()
574 auto& node_map = node_maps_[branch_index]; in ExtractBodies()
722 auto& node_map = node_maps_[branch_index]; in ExtractBodies()
734 auto& node_map = node_maps_[branch_index]; in ExtractBodies()