Searched refs:hook_function (Results 1 – 1 of 1) sorted by relevance
60 def test_custom_hook_function(self, hook_function, cell_hook_function): argument61 return hook_function, cell_hook_function74 def __init__(self, hook_function, cell_hook_function, num_class=10): argument87 self.hook = P.HookBackward(hook_function)172 net = LeNet5(hook_function=function[0], cell_hook_function=function[1])