Searched defs:bw_hook (Results 1 – 3 of 3) sorted by relevance
| /aosp_15_r20/external/pytorch/test/nn/ |
| H A D | test_module_hooks.py | 520 def bw_hook(module: nn.Module, _inputs, _outputs): function
953 def bw_hook(inc, h_module, grad_input, grad_output): function
1046 def bw_hook(module, grad_input, grad_output): function
1235 def bw_hook(inc, h_module, grad_input, grad_output): function
1336 def bw_hook(m, grad_in, grad_output): function
1601 def bw_hook(module, grad_input, grad_output): function
1616 def bw_hook(module, grad_input, grad_output): function
|
| /aosp_15_r20/external/pytorch/test/cpp/api/ |
| H A D | autograd.cpp | 983 std::function<void(int, Variable)> bw_hook( in TEST() local
|
| /aosp_15_r20/external/pytorch/test/ |
| H A D | test_autograd.py | 1741 def bw_hook(inc, grad): function
1991 def bw_hook(grad): function
|