- Sort Score
- Result 10 results
- Languages All
Results 1 - 5 of 5 for FixupSourceAndSinkEdges (0.31 sec)
-
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) -
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) -
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) -
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) -
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)