Search Options

Results per page
Sort
Preferred Languages
Advance

Results 1 - 4 of 4 for GetXlaAutoClusteringSummary (0.48 sec)

  1. tensorflow/compiler/jit/report_clustering_info_pass.cc

    namespace tensorflow {
    Status ReportClusteringInfoPass::Run(
        const GraphOptimizationPassOptions& options) {
      XlaAutoClusteringActivity activity;
      *activity.mutable_summary() = GetXlaAutoClusteringSummary(**options.graph);
      activity.set_global_jit_level(GetGlobalJitLevelForGraph(options));
      activity.set_cpu_global_jit_enabled(
          GetMarkForCompilationPassFlags()->tf_xla_cpu_global_jit);
    Registered: Sun Jun 16 05:45:23 UTC 2024
    - Last Modified: Sat Jun 15 00:14:19 UTC 2019
    - 1.3K bytes
    - Viewed (0)
  2. 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)
  3. 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)
  4. 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)
Back to top