Searched defs:backward_hook (Results 1 – 4 of 4) sorted by relevance
675 def backward_hook(grad: torch.Tensor) -> None: function711 def backward_hook(grad: torch.Tensor) -> None: function
45 def backward_hook( function
2511 def backward_hook(name, mod, grad_inp, grad_out): function
2004 def backward_hook(module, grad_input, grad_output): function