Search Options

Results per page
Sort
Preferred Languages
Advance

Results 1 - 3 of 3 for JitCompilationListener (0.45 sec)

  1. tensorflow/compiler/jit/tests/device_compiler_test_helper.h

    class DeviceCompilerSerializeTest : public ::testing::Test {
     protected:
      DeviceCompilerSerializeTest() {
        auto listener = std::make_unique<JitCompilationListener>();
        listener_ = listener.get();
        RegisterXlaActivityListener(std::move(listener));
      }
    
      JitCompilationListener* listener() const { return listener_; }
    
      // Returns a test graph that will split into two XLA clusters (due to a node
      // with _XlaCompile = false).
    Registered: Sun Jun 16 05:45:23 UTC 2024
    - Last Modified: Fri Feb 09 08:24:16 UTC 2024
    - 3.6K bytes
    - Viewed (0)
  2. tensorflow/compiler/jit/device_compiler_test.cc

        xla_device_compiler_ = CreateXlaDeviceCompiler();
        xla_device_compiler_ref_ =
            std::make_unique<core::ScopedUnref>(xla_device_compiler_);
    
        auto listener = std::make_unique<JitCompilationListener>();
        listener_ = listener.get();
        RegisterXlaActivityListener(std::move(listener));
      }
    
      XlaCompiler::Options GetDefaultXlaOptions() {
        XlaCompiler::Options options;
    Registered: Sun Jun 16 05:45:23 UTC 2024
    - Last Modified: Tue Mar 12 06:33:33 UTC 2024
    - 19.8K bytes
    - Viewed (0)
  3. tensorflow/compiler/jit/device_compilation_profiler_test.cc

    }
    
    TEST(DeviceCompilationProfilerTest, RegisterCompilation) {
      DeviceCompilationProfiler* profiler = new DeviceCompilationProfiler();
      core::ScopedUnref profiler_ref(profiler);
    
      auto listener = std::make_unique<JitCompilationListener>();
      auto listener_ptr = listener.get();
      RegisterXlaActivityListener(std::move(listener));
    
      NameAttrList function;
      function.set_name("TestFunc");
    
    Registered: Sun Jun 16 05:45:23 UTC 2024
    - Last Modified: Thu Dec 22 21:06:33 UTC 2022
    - 8.7K bytes
    - Viewed (0)
Back to top