Searched refs:hook_function (Results 1 – 2 of 2) sorted by relevance
255 def hook_function(module, input): function259 module.register_forward_pre_hook(hook_function)276 def hook_function(module, input, output): function280 module.register_forward_hook(hook_function)
554 hook_function = mock.Mock()557 callback_client.Impl(on_call_hook=hook_function),569 hook_function.assert_called_once()571 hook_function.call_args[0][0].method.full_name,