Searched refs:cell_hook_function (Results 1 – 3 of 3) sorted by relevance
| /third_party/mindspore/tests/st/pynative/ |
| D | test_pynative_hook.py | 60 def test_custom_hook_function(self, hook_function, cell_hook_function): argument
61 return hook_function, cell_hook_function
74 def __init__(self, hook_function, cell_hook_function, num_class=10): argument
80 self.conv1.register_backward_hook(cell_hook_function)
172 net = LeNet5(hook_function=function[0], cell_hook_function=function[1])
|
| /third_party/mindspore/tests/st/hook/ |
| D | test_hook_function.py | 93 def cell_hook_function(cell_id, grad_input, grad_output): function
103 self.relu.register_backward_hook(cell_hook_function)
116 self.relu.register_backward_hook(cell_hook_function)
|
| /third_party/mindspore/tests/ut/python/pynative_mode/ |
| D | test_hook.py | 54 def cell_hook_function(cell_id, grad_input, grad_output): function
102 self.conv2.register_backward_hook(cell_hook_function)
|