Search Options

Results per page
Sort
Preferred Languages
Advance

Results 1 - 10 of 15 for FixupSourceAndSinkEdges (0.49 sec)

  1. tensorflow/compiler/jit/xla_cluster_util_test.cc

      Output enter =
          ops::internal::Enter(root.WithOpName("enter"), a, "only_frame");
      Output exit = ops::internal::Exit(root.WithOpName("exit"), enter);
      Output b = ops::Add(root.WithOpName("b"), a, exit);
    
      FixupSourceAndSinkEdges(root.graph());
    
      GraphCycles cycles;
      TF_ASSERT_OK(CreateCycleDetectionGraph(root.graph(), &cycles).status());
      EXPECT_FALSE(cycles.CanContractEdge(a.node()->id(), b.node()->id()));
    }
    
    Registered: Sun Jun 16 05:45:23 UTC 2024
    - Last Modified: Wed Feb 21 09:53:30 UTC 2024
    - 10.8K bytes
    - Viewed (0)
  2. tensorflow/compiler/jit/deadness_analysis_test.cc

    using deadness_analysis_internal::ComputePredicates;
    using deadness_analysis_internal::PredicateMapTy;
    
    Status AnalyzeDeadness(Graph* graph,
                           std::unique_ptr<DeadnessAnalysis>* result) {
      FixupSourceAndSinkEdges(graph);
      return DeadnessAnalysis::Run(*graph, result);
    }
    
    ops::Switch CreateSwitch(const Scope& root, const string& prefix) {
      Output value = ops::Placeholder(root.WithOpName(prefix + "/value"), DT_FLOAT);
    Registered: Sun Jun 16 05:45:23 UTC 2024
    - Last Modified: Thu Feb 22 06:59:07 UTC 2024
    - 51.6K bytes
    - Viewed (0)
  3. tensorflow/compiler/jit/cluster_scoping_pass_test.cc

    #include "tensorflow/core/platform/test.h"
    #include "tensorflow/core/public/session_options.h"
    
    namespace tensorflow {
    namespace {
    
    Status ClusterScoping(std::unique_ptr<Graph>* graph) {
      FixupSourceAndSinkEdges(graph->get());
    
      GraphOptimizationPassWrapper wrapper;
      wrapper.session_options.config.mutable_graph_options()
          ->mutable_optimizer_options()
          ->set_global_jit_level(OptimizerOptions::ON_2);
    Registered: Sun Jun 16 05:45:23 UTC 2024
    - Last Modified: Wed Apr 29 16:20:48 UTC 2020
    - 6.7K bytes
    - Viewed (0)
  4. tensorflow/compiler/jit/xla_compile_util.cc

                            .Attr("T", result_types[i])
                            .Attr("index", i)
                            .Finalize(graph.get(), &node);
        TF_RETURN_IF_ERROR(status);
      }
      FixupSourceAndSinkEdges(graph.get());
      return graph;
    }
    
    bool UsePjRtForSingleDeviceCompilation(const DeviceType& device_type) {
      const auto& rollout_config = GetXlaOpsCommonFlags()->tf_xla_use_device_api;
    Registered: Sun Jun 16 05:45:23 UTC 2024
    - Last Modified: Wed Feb 21 09:53:30 UTC 2024
    - 4.6K bytes
    - Viewed (0)
  5. tensorflow/compiler/jit/build_xla_ops_pass_test.cc

        if (n->requested_device().empty()) {
          n->set_assigned_device_name(kCpuDevice);
        } else {
          n->set_assigned_device_name(n->requested_device());
        }
      }
    
      FixupSourceAndSinkEdges(graph.get());
    
      GraphOptimizationPassWrapper wrapper;
      GraphOptimizationPassOptions opt_options =
          wrapper.CreateGraphOptimizationPassOptions(&graph);
      opt_options.flib_def = &flib_def;
    
    Registered: Sun Jun 16 05:45:23 UTC 2024
    - Last Modified: Thu Feb 22 08:47:20 UTC 2024
    - 12.2K bytes
    - Viewed (0)
  6. tensorflow/compiler/jit/increase_dynamism_for_auto_jit_pass.cc

        TF_RETURN_IF_ERROR(
            RewriteSlice(g, n, *slice_inputs, *GetXlaClusterForNode(*n)));
      }
    
      if (!slices_to_rewrite.empty()) {
        // We've added constants to the graph; hook them up to _SOURCE.
        FixupSourceAndSinkEdges(g);
      }
    
      *changed = !slices_to_rewrite.empty();
    
      return absl::OkStatus();
    }
    }  // namespace
    
    Status IncreaseDynamismForAutoJitPass::Run(
    Registered: Sun Jun 16 05:45:23 UTC 2024
    - Last Modified: Tue Mar 12 06:33:33 UTC 2024
    - 13.8K bytes
    - Viewed (0)
  7. tensorflow/compiler/jit/resource_operation_safety_analysis_test.cc

      return ops::Const(scope.WithOpName("Const" + id), 42.0f).node();
    }
    
    Status ComputeIncompatiblePairs(Graph* g,
                                    std::vector<std::pair<int, int>>* result) {
      FixupSourceAndSinkEdges(g);
      return ComputeIncompatibleResourceOperationPairs(*g, &g->flib_def(), {},
                                                       result);
    }
    
    TEST(ResourceOperationSafetyAnalysisTest, WriteRead) {
    Registered: Sun Jun 16 05:45:23 UTC 2024
    - Last Modified: Tue Apr 28 16:53:59 UTC 2020
    - 18.7K bytes
    - Viewed (0)
  8. tensorflow/compiler/jit/partially_decluster_pass_test.cc

    REGISTER_KERNEL_BUILDER(
        Name("FakeResourceUpdate").Device(DEVICE_CPU).HostMemory("something_else"),
        FakeResourceUpdateOp);
    
    Status PartiallyDecluster(std::unique_ptr<Graph>* graph) {
      FixupSourceAndSinkEdges(graph->get());
      // Assign all nodes to the CPU device.
      static const char* kCpuDevice = "/job:localhost/replica:0/task:0/cpu:0";
      for (Node* n : (*graph)->nodes()) {
        if (n->assigned_device_name().empty()) {
    Registered: Sun Jun 16 05:45:23 UTC 2024
    - Last Modified: Fri Jun 10 12:32:39 UTC 2022
    - 23K bytes
    - Viewed (0)
  9. tensorflow/compiler/jit/mark_for_compilation_pass_test.cc

      Scope root = Scope::NewRootScope().ExitOnError();
    
      Node* read = MakeRead(root, "R");
      Node* write = MakeWrite(root, "W");
    
      root.graph()->AddControlEdge(read, write);
    
      FixupSourceAndSinkEdges(root.graph());
      std::unique_ptr<Graph> graph(new Graph(OpRegistry::Global()));
      TF_EXPECT_OK(root.ToGraph(graph.get()));
      TF_ASSERT_OK(MarkForCompilationPassTestHelper::MarkForCompilation(&graph));
    Registered: Sun Jun 16 05:45:23 UTC 2024
    - Last Modified: Wed Feb 14 10:11:10 UTC 2024
    - 79.6K bytes
    - Viewed (0)
  10. tensorflow/compiler/jit/extract_outside_compilation_pass.cc

        // We use ReverseDFS() to copy nodes. Make sure all nodes are reverse
        // reachable from sink node so all nodes will be copied.
        // TODO(b/77601805): consolidate copy graph functions.
        FixupSourceAndSinkEdges(host_fbody->graph);
    
        std::map<const Node*, Node*> node_map;
        node_map[host_fbody->graph->source_node()] = (*host_graph)->source_node();
    Registered: Sun Jun 16 05:45:23 UTC 2024
    - Last Modified: Tue Mar 12 06:33:33 UTC 2024
    - 104.7K bytes
    - Viewed (0)
Back to top