(self, module, inp, out)
| 89 | self.hook = module.register_forward_hook(self.forward_hook) |
| 90 | |
| 91 | def forward_hook(self, module, inp, out): |
| 92 | self.input = inp |
| 93 | self.output = out |
| 94 | self.output.register_hook(self.backward_hook) |
| 95 | |
| 96 | def backward_hook(self, grad): |
| 97 | self.grad = grad |
nothing calls this directly
no outgoing calls
no test coverage detected