Search Options

Results per page
Sort
Preferred Languages
Advance

Results 11 - 15 of 15 for FixupSourceAndSinkEdges (0.52 sec)

  1. tensorflow/compiler/mlir/tensorflow/translate/export_graphdef.cc

      for (Operation& inst : graph_op.GetBody()) {
        TF_RETURN_IF_ERROR(exporter.AddEdge(&inst));
      }
      // Fixes the edges between the inserted nodes and special "_SOURCE" and
      // "_SINK".
      FixupSourceAndSinkEdges(graph.get());
    
      FixupInputNamesFromEdges(graph.get());
    
      TF_RETURN_IF_ERROR(
          exporter.GetControlRetNodes(graph_op.GetFetch(), control_ret_nodes));
    
      return graph;
    }
    
    Registered: Sun Jun 16 05:45:23 UTC 2024
    - Last Modified: Wed May 01 11:17:36 UTC 2024
    - 35.2K bytes
    - Viewed (0)
  2. tensorflow/compiler/mlir/tf2xla/api/v2/tf_executor_to_graph.cc

      for (Operation& inst : graph_op.GetBody()) {
        TF_RETURN_IF_ERROR(exporter.AddEdge(&inst));
      }
      // Fixes the edges between the inserted nodes and special "_SOURCE" and
      // "_SINK".
      FixupSourceAndSinkEdges(graph.get());
    
      FixupInputNamesFromEdges(graph.get());
    
      TF_RETURN_IF_ERROR(
          exporter.GetControlRetNodes(graph_op.GetFetch(), control_ret_nodes));
    
      return graph;
    }
    
    Registered: Sun Jun 16 05:45:23 UTC 2024
    - Last Modified: Thu Jun 06 23:04:51 UTC 2024
    - 35.2K bytes
    - Viewed (0)
  3. tensorflow/compiler/jit/encapsulate_subgraphs_pass.cc

      TF_RETURN_IF_ERROR(CopySubgraphEdges(node_images, &src_arg_pairs));
      MarkGuaranteedConstants(*graph_in_, src_arg_pairs);
    
      for (auto& entry : subgraphs_) {
        Subgraph& subgraph = entry.second;
        FixupSourceAndSinkEdges(subgraph.GetGraph());
      }
    
      if (VLOG_IS_ON(1)) {
        // Dump subgraphs.
        for (auto& entry : subgraphs_) {
          DumpGraphToFile(
    Registered: Sun Jun 16 05:45:23 UTC 2024
    - Last Modified: Thu Feb 22 08:47:20 UTC 2024
    - 51K bytes
    - Viewed (0)
  4. tensorflow/compiler/jit/mark_for_compilation_pass.cc

      // connected but sometimes the incoming graph does not follow this invariant.
      // So fix up the source and sink edges before calling into deadness analysis.
      FixupSourceAndSinkEdges(graph);
    
      for (Node* n : graph->nodes()) {
        // See explanation on `kXlaAlreadyClustered`.
        if (n->attrs().Find(kXlaAlreadyClustered)) {
          return absl::OkStatus();
        }
    Registered: Sun Jun 16 05:45:23 UTC 2024
    - Last Modified: Wed Feb 21 12:19:41 UTC 2024
    - 85.3K bytes
    - Viewed (0)
  5. tensorflow/compiler/mlir/tensorflow/translate/import_model.cc

        }
      }
    
      // Since we might have inserted and removed nodes from the graph, fix
      // source/sink edges and reconstruct the RPO ordering of nodes
      FixupSourceAndSinkEdges(graph_.get());
    
      // Prune nodes in the graph that are not reachable from the output.
      if (specs_.prune_unused_nodes) {
        std::unordered_set<const Node*> prune_start;
    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