(self, x, y)
| 69 | return all_alphas |
| 70 | |
| 71 | def _loss(self, x, y): |
| 72 | logits = tf.cast(x, dtype=tf.float32) |
| 73 | loss = self._loss_fn_with_scan(logits, self.transition_param) |
| 74 | return tf.reduce_mean(loss) |
| 75 | |
| 76 | @tf.function |
| 77 | def train_on_batch(self, *args): |
no test coverage detected