- Sort Score
- Result 10 results
- Languages All
Results 1 - 4 of 4 for GetNodesRelatedToRefVariables (0.3 sec)
-
tensorflow/compiler/jit/xla_cluster_util.cc
// size of `callee_ref_nodes` but for now we don't ceare. if (!callee_ref_nodes.empty()) { return true; } } return false; } // Helper for GetNodesRelatedToRefVariables that traverses the graph in one // direction. Status GetNodesRelatedToRefVariablesInDirection( const Graph& graph, FunctionLibraryRuntime* lib_runtime,
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Thu Feb 29 08:39:39 UTC 2024 - 21.3K bytes - Viewed (0) -
tensorflow/compiler/jit/xla_cluster_util.h
// // We assume each node has a trivial path to itself so the returned set includes // all of the nodes that have ref variables as input or output. absl::StatusOr<absl::flat_hash_set<Node*>> GetNodesRelatedToRefVariables( const Graph& graph, FunctionLibraryRuntime* lib_runtime); // Deterministically serialized the graph to a byte string. absl::StatusOr<std::string> SerializeGraphDeterministic(const Graph& graph);
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/xla_cluster_util_test.cc
FunctionLibraryRuntime* lib_runtime = pflr->GetFLR(ProcessFunctionLibraryRuntime::kDefaultFLRDevice); TF_ASSIGN_OR_RETURN(absl::flat_hash_set<Node*> nodes_related_to_ref_vars, GetNodesRelatedToRefVariables(*graph, lib_runtime)); std::vector<string> names; absl::c_transform(nodes_related_to_ref_vars, std::back_inserter(names), [](Node* n) { return n->name(); }); absl::c_sort(names);
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/encapsulate_subgraphs_pass.cc
options.flib_def); } *options.graph = std::move(graph_out); TF_ASSIGN_OR_RETURN(absl::flat_hash_set<Node*> ref_related_nodes, GetNodesRelatedToRefVariables(**options.graph, flr)); for (Node* node : (*options.graph)->nodes()) { bool has_ref_vars = ref_related_nodes.contains(node); node->AddAttr(kXlaHasReferenceVarsAttr, has_ref_vars);
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Thu Feb 22 08:47:20 UTC 2024 - 51K bytes - Viewed (0)