(self, grad, loss_scale_reciprical)
| 611 | ] |
| 612 | |
| 613 | def _scale_grad(self, grad, loss_scale_reciprical): |
| 614 | if isinstance(grad, ops.IndexedSlices): |
| 615 | grad_vals = grad.values * loss_scale_reciprical |
| 616 | return ops.IndexedSlices(grad_vals, grad.indices, grad.dense_shape) |
| 617 | return grad * loss_scale_reciprical |
| 618 | |
| 619 | def apply_gradients(self, grads_and_vars, global_step=None, name=None): |
| 620 | """Apply gradients to variables. |