Searched refs:wrapped_vars (Results 1 – 1 of 1) sorted by relevance
| /external/tensorflow/tensorflow/python/keras/mixed_precision/ |
| D | loss_scale_optimizer.py | 729 wrapped_vars = _UnwrapPreventer([v for _, v in grads_and_vars])
732 args=(grads, wrapped_vars, name, experimental_aggregate_gradients))
748 def _apply_gradients(self, grads, wrapped_vars, name, argument
754 list(zip(grads, wrapped_vars.value)), name,
|