Home
last modified time | relevance | path

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

/external/tensorflow/tensorflow/python/tpu/
Dtpu_optimizer.py163 def apply_gradients(self, grads_and_vars, global_step=None, name=None): member in CrossShardOptimizer
Dtpu_embedding_v2.py577 def apply_gradients(self, gradients, name: Text = None): member in TPUEmbedding
/external/tensorflow/tensorflow/python/training/
Dsync_replicas_optimizer.py224 def apply_gradients(self, grads_and_vars, global_step=None, name=None): member in SyncReplicasOptimizer
Doptimizer.py539 def apply_gradients(self, grads_and_vars, global_step=None, name=None): member in Optimizer
/external/tensorflow/tensorflow/python/training/experimental/
Dloss_scale_optimizer.py156 def apply_gradients(self, grads_and_vars, global_step=None, name=None): member in MixedPrecisionLossScaleOptimizer
/external/tensorflow/tensorflow/python/eager/benchmarks/resnet50/
Dresnet50_test.py54 def apply_gradients(model, optimizer, gradients): function
/external/tensorflow/tensorflow/python/keras/mixed_precision/
Dloss_scale_optimizer_test.py471 def apply_gradients(self, member in LossScaleOptimizerTest.testApplyGradientsGetsUnwrappedTensors.MyOptimizer
Dloss_scale_optimizer.py699 def apply_gradients(self, member in LossScaleOptimizer
/external/tensorflow/tensorflow/python/keras/
Doptimizer_v1.py786 def apply_gradients(self, grads_and_vars): member in TFOptimizer
/external/tensorflow/tensorflow/python/keras/optimizer_v2/
Doptimizer_v2.py592 def apply_gradients(self, member in OptimizerV2
/external/tensorflow/tensorflow/python/keras/engine/
Dtraining_test.py1492 … def apply_gradients(self, grads_and_vars, name=None): # pylint: disable=useless-super-delegation member in TrainingTest.test_calling_aggregate_gradient._OptimizerOverrideApplyGradients