(self, hidden_states)
| 644 | return not should_calc |
| 645 | |
| 646 | def store(self, hidden_states): |
| 647 | self.previous_residual = hidden_states - self.previous_hidden_states |
| 648 | self.previous_hidden_states = None |
| 649 | |
| 650 | def update(self, hidden_states): |
| 651 | hidden_states = hidden_states + self.previous_residual |