Search Options

Results per page
Sort
Preferred Languages
Advance

Results 1 - 2 of 2 for sink_node (0.08 sec)

  1. tensorflow/compiler/jit/extract_outside_compilation_pass.cc

        FixupSourceAndSinkEdges(host_fbody->graph);
    
        std::map<const Node*, Node*> node_map;
        node_map[host_fbody->graph->source_node()] = (*host_graph)->source_node();
        node_map[host_fbody->graph->sink_node()] = (*host_graph)->sink_node();
        Status s;
        ReverseDFS(
            *host_fbody->graph, /*enter=*/nullptr,
            [&](const Node* n) {
              if (!s.ok()) {
                return;
              }
    
    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/mlir/tensorflow/translate/import_model.cc

                                     true /* skip test for duplicates */);
            }
            if (placeholder_node->out_edges().empty()) {
              graph_->AddControlEdge(placeholder_node, graph_->sink_node(),
                                     true /* skip test for duplicates */);
            }
            remapped_feeds_[{it.first, index}] = placeholder_node->name();
    Registered: Sun Jun 16 05:45:23 UTC 2024
    - Last Modified: Wed May 01 11:17:36 UTC 2024
    - 183.2K bytes
    - Viewed (0)
Back to top