Search Options

Results per page
Sort
Preferred Languages
Advance

Results 1 - 2 of 2 for out_edge_info (0.34 sec)

  1. 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)
  2. 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)
Back to top