Searched refs:callback_registry (Results 1 – 2 of 2) sorted by relevance
326 auto* callback_registry = DebugCallbackRegistry::singleton(); in TEST_F() local327 callback_registry->RegisterCallback( in TEST_F()344 callback_registry->UnregisterCallback("test_callback"); in TEST_F()
458 auto* callback_registry = DebugCallbackRegistry::singleton(); in PublishDebugTensor() local459 auto* callback = callback_registry->GetCallback(dump_root_dir); in PublishDebugTensor()