Home
last modified time | relevance | path

Searched defs:wrapped_vars (Results 1 – 1 of 1) sorted by relevance

/external/tensorflow/tensorflow/python/keras/mixed_precision/
Dloss_scale_optimizer.py645 def _apply_gradients_cross_replica(distribution, grads, wrapped_vars, argument
665 def _apply_gradients(self, grads, wrapped_vars, name): argument