Home
last modified time | relevance | path

Searched refs:callback_registry (Results 1 – 2 of 2) sorted by relevance

/external/tensorflow/tensorflow/core/debug/
Ddebug_io_utils_test.cc326 auto* callback_registry = DebugCallbackRegistry::singleton(); in TEST_F() local
327 callback_registry->RegisterCallback( in TEST_F()
344 callback_registry->UnregisterCallback("test_callback"); in TEST_F()
Ddebug_io_utils.cc458 auto* callback_registry = DebugCallbackRegistry::singleton(); in PublishDebugTensor() local
459 auto* callback = callback_registry->GetCallback(dump_root_dir); in PublishDebugTensor()