- Sort Score
- Result 10 results
- Languages All
Results 1 - 2 of 2 for GetNodesRelatedToRefVariables (0.32 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_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)