- Sort Score
- Result 10 results
- Languages All
Results 1 - 3 of 3 for GetXlaAutoClusteringSummary (0.19 sec)
-
tensorflow/compiler/jit/xla_cluster_util.h
// not the data itself. bool IsShapeConsumerOp(const Node& node); // Computes a clustering summary for `graph`. See documentation on // `XlaAutoClusteringSummary` for details. XlaAutoClusteringSummary GetXlaAutoClusteringSummary(const Graph& graph); // Returns the set of nodes that have a path to or from nodes that may have ref // variables as input or output. //
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.cc
result->set_size(info.size); HistogramMapToRepeatedOpAndCount(result->mutable_op_histogram(), info.op_histogram); } } // namespace XlaAutoClusteringSummary GetXlaAutoClusteringSummary(const Graph& graph) { absl::flat_hash_map<absl::string_view, ClusterInfo> cluster_name_to_info; XlaAutoClusteringSummary result; absl::flat_hash_map<absl::string_view, int> unclustered_op_histogram;
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/mark_for_compilation_pass.cc
} void MarkForCompilationPassImpl::VLogClusteringSummary() { if (!VLOG_IS_ON(2)) { return; } XlaAutoClusteringSummary auto_clustering_info = GetXlaAutoClusteringSummary(*graph_); VLOG(2) << "*** Clustering info for graph of size " << graph_->num_nodes(); VLOG(2) << " Built " << auto_clustering_info.clusters_size() << " clusters, size "
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Wed Feb 21 12:19:41 UTC 2024 - 85.3K bytes - Viewed (0)