- Sort Score
- Result 10 results
- Languages All
Results 1 - 2 of 2 for out_edge_info (0.12 sec)
-
tensorflow/compiler/jit/extract_outside_compilation_pass.cc
// This is to avoid multiple producers. std::vector<OutEdgeInfo> out_edge_info; out_edge_info.reserve(n->out_edges().size()); for (auto edge : n->out_edges()) { out_edge_info.push_back( {edge->dst(), edge->src_output(), edge->dst_input()}); } g->RemoveNode(n); for (const OutEdgeInfo& edge : out_edge_info) { if (edge.dst_input == Graph::kControlSlot) {
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Tue Mar 12 06:33:33 UTC 2024 - 104.7K bytes - Viewed (0) -
tensorflow/compiler/jit/encapsulate_util.cc
node_name); } // Change all usage node to use the original node instead. Node* original_node = iter->second; std::vector<const Edge*> control_edges; std::vector<OutEdgeInfo> data_edges; for (auto e : n->out_edges()) { if (e->IsControlEdge()) { control_edges.push_back(e); } else { data_edges.push_back({e->dst(), e->src_output(), e->dst_input()});
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Tue Mar 12 06:33:33 UTC 2024 - 15.5K bytes - Viewed (0)