- Sort Score
- Result 10 results
- Languages All
Results 1 - 6 of 6 for RegisterXlaActivityListener (0.31 sec)
-
tensorflow/compiler/jit/xla_activity_logging_listener.cc
bool enabled_by_default = true; return enabled_by_default; } return absl::string_view(log_xla_activity) == "1"; } }; bool Register() { RegisterXlaActivityListener(std::make_unique<XlaActivityLoggingListener>()); return false; } bool unused = Register(); } // namespace
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Thu Feb 22 08:47:20 UTC 2024 - 2.2K bytes - Viewed (0) -
tensorflow/compiler/jit/xla_activity_listener.cc
XlaOptimizationRemark remark; remark.set_warning(optimization_warning); remark.set_debug_information(std::move(debug_information)); return BroadcastOptimizationRemark(std::move(remark)); } void RegisterXlaActivityListener( std::unique_ptr<XlaActivityListener> listener) { XlaActivityListenerList* listener_list = GetXlaActivityListenerList(); absl::WriterMutexLock writer_lock(&listener_list->mutex);
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Thu Feb 22 08:47:20 UTC 2024 - 3.3K bytes - Viewed (0) -
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/xla_activity_listener_test.cc
}; class XlaActivityListenerTest : public ::testing::Test { protected: XlaActivityListenerTest() { auto listener = std::make_unique<TestListener>(); listener_ = listener.get(); RegisterXlaActivityListener(std::move(listener)); } TestListener* listener() const { return listener_; } private: TestListener* listener_; }; GraphDef CreateGraphDef() {
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Thu Feb 22 08:47:20 UTC 2024 - 5.9K bytes - Viewed (0) -
tensorflow/compiler/jit/device_compilation_profiler_test.cc
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"); std::vector<XlaJitCompilationActivity> expected_activities; for (int i = 0; i < 5; ++i) {
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Thu Dec 22 21:06:33 UTC 2022 - 8.7K bytes - Viewed (0) -
tensorflow/compiler/jit/device_compiler_test.cc
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; options.device_type = DeviceType(DEVICE_GPU_XLA_JIT);
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Tue Mar 12 06:33:33 UTC 2024 - 19.8K bytes - Viewed (0)