- Sort Score
- Result 10 results
- Languages All
Results 1 - 3 of 3 for JitCompilationListener (0.27 sec)
-
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) -
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) -
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)