- Sort Score
- Result 10 results
- Languages All
Results 1 - 10 of 15 for FixupSourceAndSinkEdges (0.49 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/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) -
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) -
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) -
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/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) -
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) -
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) -
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)