| /external/tensorflow/tensorflow/python/keras/optimizer_v2/ |
| D | utils.py | 28 def all_reduce_sum_gradients(grads_and_vars): argument
62 def filter_empty_gradients(grads_and_vars): argument
92 def gradient_clipnorm_fn(grads_and_vars): argument
113 def gradient_clipnorm_fn(grads_and_vars): argument
134 def gradient_clipvalue_fn(grads_and_vars): argument
150 def _all_reduce_sum_fn(distribution, grads_and_vars): argument
|
| D | optimizer_v2.py | 479 def _transform_unaggregated_gradients(self, grads_and_vars): argument
483 def _aggregate_gradients(self, grads_and_vars): argument
487 def _transform_gradients(self, grads_and_vars): argument
593 grads_and_vars, argument
676 def _distributed_apply(self, distribution, grads_and_vars, name, apply_state): argument
|
| /external/tensorflow/tensorflow/python/training/experimental/ |
| D | loss_scale_optimizer.py | 156 def apply_gradients(self, grads_and_vars, global_step=None, name=None): argument
193 grads_and_vars, argument
228 def _apply_gradients(self, distribution, grads_and_vars, global_step, name): argument
|
| /external/tensorflow/tensorflow/python/tpu/ |
| D | tpu_optimizer.py | 163 def apply_gradients(self, grads_and_vars, global_step=None, name=None): argument
|
| /external/tensorflow/tensorflow/python/training/ |
| D | sync_replicas_optimizer.py | 224 def apply_gradients(self, grads_and_vars, global_step=None, name=None): argument
|
| D | optimizer.py | 539 def apply_gradients(self, grads_and_vars, global_step=None, name=None): argument
652 grads_and_vars, argument
|
| /external/tensorflow/tensorflow/python/keras/mixed_precision/ |
| D | loss_scale_optimizer.py | 700 grads_and_vars, argument
714 def _apply_gradients_cross_replica(self, distribution, grads_and_vars, name, argument
864 def _aggregate_gradients(self, grads_and_vars): argument
|
| D | loss_scale_optimizer_test.py | 472 grads_and_vars, argument
|
| /external/tensorflow/tensorflow/python/keras/ |
| D | optimizer_v1.py | 786 def apply_gradients(self, grads_and_vars): argument
|
| /external/tensorflow/tensorflow/python/keras/distribute/ |
| D | distribute_strategy_test.py | 2021 def custom_transform(grads_and_vars): argument
|
| /external/tensorflow/tensorflow/python/keras/engine/ |
| D | training_test.py | 1492 … def apply_gradients(self, grads_and_vars, name=None): # pylint: disable=useless-super-delegation argument
|