Searched refs:DebugCallbackRegistry (Results 1 – 5 of 5) sorted by relevance
20 DebugCallbackRegistry::DebugCallbackRegistry() {} in DebugCallbackRegistry() function in tensorflow::DebugCallbackRegistry22 /*static */ DebugCallbackRegistry* DebugCallbackRegistry::instance_ = nullptr;24 DebugCallbackRegistry* DebugCallbackRegistry::singleton() { in singleton()26 instance_ = new DebugCallbackRegistry(); in singleton()31 void DebugCallbackRegistry::RegisterCallback(const string& key, in RegisterCallback()37 DebugCallbackRegistry::EventCallback* DebugCallbackRegistry::GetCallback( in GetCallback()44 void DebugCallbackRegistry::UnregisterCallback(const string& key) { in UnregisterCallback()
40 class DebugCallbackRegistry {45 static DebugCallbackRegistry* singleton();58 DebugCallbackRegistry();66 static DebugCallbackRegistry* instance_;
326 auto* callback_registry = DebugCallbackRegistry::singleton(); in TEST_F()
457 auto* callback_registry = DebugCallbackRegistry::singleton(); in PublishDebugTensor()
36 // callback registered with the DebugCallbackRegistry for event_key.