Searched refs:_backward (Results 1 – 2 of 2) sorted by relevance
221 out = self._backward(y1, y2)364 def _backward(self, y1, y2): member in RevBlock
988 def _backward(accs, elems): function1012 bwd, cum_log_sum = _scan(_backward, (observed_log_probs, mask),