(self, grads)
| 603 | return loss_value |
| 604 | |
| 605 | def _unscale_grads(self, grads): |
| 606 | loss_scale = self._loss_scale() |
| 607 | loss_scale_reciprical = 1 / loss_scale |
| 608 | return [ |
| 609 | None if g is None else self._scale_grad(g, loss_scale_reciprical) |
| 610 | for g in grads |
| 611 | ] |
| 612 | |
| 613 | def _scale_grad(self, grad, loss_scale_reciprical): |
| 614 | if isinstance(grad, ops.IndexedSlices): |
no test coverage detected