Search Options

Results per page
Sort
Preferred Languages
Advance

Results 1 - 7 of 7 for XlaAutoClusteringActivity (0.38 sec)

  1. tensorflow/compiler/jit/xla_activity_logging_listener.cc

    class XlaActivityLoggingListener final : public XlaActivityListener {
     public:
      Status Listen(
          const XlaAutoClusteringActivity& auto_clustering_activity) override {
        if (!IsEnabled()) {
          VLOG(3) << "Logging XlaAutoClusteringActivity disabled";
          return absl::OkStatus();
        }
    
        return absl::OkStatus();
      }
    
      Status Listen(
    Registered: Sun Jun 16 05:45:23 UTC 2024
    - Last Modified: Thu Feb 22 08:47:20 UTC 2024
    - 2.2K bytes
    - Viewed (0)
  2. tensorflow/compiler/jit/xla_activity_listener_test.cc

        return absl::OkStatus();
      }
    
      ~TestListener() override {}
    
      const XlaAutoClusteringActivity& auto_clustering_activity() const {
        return auto_clustering_activity_;
      }
      const XlaJitCompilationActivity& jit_compilation_activity() const {
        return jit_compilation_activity_;
      }
    
     private:
      XlaAutoClusteringActivity auto_clustering_activity_;
      XlaJitCompilationActivity jit_compilation_activity_;
    };
    
    Registered: Sun Jun 16 05:45:23 UTC 2024
    - Last Modified: Thu Feb 22 08:47:20 UTC 2024
    - 5.9K bytes
    - Viewed (0)
  3. tensorflow/compiler/jit/xla_activity_listener.h

    #include "tensorflow/compiler/jit/xla_activity.pb.h"
    #include "tensorflow/core/lib/core/status.h"
    
    namespace tensorflow {
    // Broadcast `auto_clustering_activity` to all the registered listeners.
    Status BroadcastXlaActivity(XlaAutoClusteringActivity auto_clustering_activity);
    
    // Broadcast `jit_compilation_activity` to all the registered listeners.
    Status BroadcastXlaActivity(XlaJitCompilationActivity jit_compilation_activity);
    
    Registered: Sun Jun 16 05:45:23 UTC 2024
    - Last Modified: Wed Aug 07 11:04:01 UTC 2019
    - 2.9K bytes
    - Viewed (0)
  4. tensorflow/compiler/jit/report_clustering_info_pass.cc

    #include "tensorflow/compiler/jit/xla_cluster_util.h"
    
    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(
    Registered: Sun Jun 16 05:45:23 UTC 2024
    - Last Modified: Sat Jun 15 00:14:19 UTC 2019
    - 1.3K bytes
    - Viewed (0)
  5. tensorflow/compiler/jit/xla_activity_listener.cc

      Status s = ForEachListener([](XlaActivityListener* listener) {
        listener->Flush();
        return absl::OkStatus();
      });
      CHECK(s.ok());
    }
    }  // namespace
    
    Status BroadcastXlaActivity(
        XlaAutoClusteringActivity auto_clustering_activity) {
      return ForEachListener([&](XlaActivityListener* listener) {
        return listener->Listen(auto_clustering_activity);
      });
    }
    
    Status BroadcastXlaActivity(
    Registered: Sun Jun 16 05:45:23 UTC 2024
    - Last Modified: Thu Feb 22 08:47:20 UTC 2024
    - 3.3K bytes
    - Viewed (0)
  6. tensorflow/compiler/jit/tests/device_compiler_test_helper.h

    namespace tensorflow {
    
    // A listener to inspect the use of XLA's persistent compilation cache entries.
    class JitCompilationListener : public XlaActivityListener {
     public:
      Status Listen(
          const XlaAutoClusteringActivity& auto_clustering_activity) override {
        return absl::OkStatus();
      }
    
      Status Listen(
          const XlaJitCompilationActivity& jit_compilation_activity) override {
    Registered: Sun Jun 16 05:45:23 UTC 2024
    - Last Modified: Fri Feb 09 08:24:16 UTC 2024
    - 3.6K bytes
    - Viewed (0)
  7. tensorflow/compiler/jit/xla_activity.proto

      repeated OpAndCount unclustered_op_histogram = 4;
    }
    
    // Listeners listening for auto clustering events get messages of this type.
    //
    // Next ID: 4
    message XlaAutoClusteringActivity {
      // The value of GlobalJitLevel, as determined by `GetGlobalJitLevelForGraph`.
      // This determines if global auto-clustering is enabled.
      OptimizerOptions.GlobalJitLevel global_jit_level = 1;
    
    Registered: Sun Jun 16 05:45:23 UTC 2024
    - Last Modified: Tue Mar 15 03:11:33 UTC 2022
    - 3.6K bytes
    - Viewed (0)
Back to top