Searched refs:backward_pre_hook (Results 1 – 2 of 2) sorted by relevance
40 def backward_pre_hook( function56 module.register_full_backward_pre_hook(backward_pre_hook)
2520 def backward_pre_hook(name, mod, grad_out): function2530 partial(backward_pre_hook, name)