Searched defs:apply_gradients (Results 1 – 9 of 9) sorted by relevance
| /external/tensorflow/tensorflow/python/tpu/ |
| D | tpu_optimizer.py | 164 def apply_gradients(self, grads_and_vars, global_step=None, name=None): member in CrossShardOptimizer
|
| D | tpu_embedding_v2.py | 614 def apply_gradients(self, gradients, name: Optional[Text] = None): member in TPUEmbedding
|
| /external/tensorflow/tensorflow/python/training/ |
| D | sync_replicas_optimizer.py | 221 def apply_gradients(self, grads_and_vars, global_step=None, name=None): member in SyncReplicasOptimizer
|
| D | optimizer.py | 623 def apply_gradients(self, grads_and_vars, global_step=None, name=None): member in Optimizer
|
| /external/tensorflow/tensorflow/python/training/experimental/ |
| D | loss_scale_optimizer.py | 153 def apply_gradients(self, grads_and_vars, global_step=None, name=None): member in MixedPrecisionLossScaleOptimizer
|
| /external/tensorflow/tensorflow/python/eager/benchmarks/resnet50/ |
| D | resnet50_test.py | 49 def apply_gradients(model, optimizer, gradients): function
|
| /external/tensorflow/tensorflow/python/keras/ |
| D | optimizer_v1.py | 802 def apply_gradients(self, grads_and_vars): member in TFOptimizer
|
| /external/tensorflow/tensorflow/python/keras/mixed_precision/ |
| D | loss_scale_optimizer.py | 591 def apply_gradients(self, member in LossScaleOptimizer
|
| /external/tensorflow/tensorflow/python/keras/optimizer_v2/ |
| D | optimizer_v2.py | 594 def apply_gradients(self, member in OptimizerV2
|