Searched defs:cell_hook_function (Results 1 – 3 of 3) sorted by relevance
60 def test_custom_hook_function(self, hook_function, cell_hook_function): argument74 def __init__(self, hook_function, cell_hook_function, num_class=10): argument
93 def cell_hook_function(cell_id, grad_input, grad_output): function
54 def cell_hook_function(cell_id, grad_input, grad_output): function