Home
last modified time | relevance | path

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

/external/tensorflow/tensorflow/contrib/mixed_precision/python/
Dloss_scale_optimizer.py114 def compute_gradients(self, member in LossScaleOptimizer
/external/tensorflow/tensorflow/contrib/opt/python/training/
Dmoving_average_optimizer.py90 def compute_gradients(self, *args, **kwargs): member in MovingAverageOptimizer
Ddrop_stale_gradient_optimizer.py63 def compute_gradients(self, loss, *args, **kwargs): member in DropStaleGradientOptimizer
Dmoving_average_optimizer_test.py237 def compute_gradients(self, *args, **kwargs): member in MovingAverageOptimizerTest.testCorrectOverride.WrapperOptimizer
Dvariable_clipping_optimizer.py83 def compute_gradients(self, *args, **kwargs): member in VariableClippingOptimizer
Dmodel_average_optimizer.py149 def compute_gradients(self, *args, **kwargs): member in ModelAverageOptimizer
Delastic_average_optimizer.py213 def compute_gradients(self, member in ElasticAverageOptimizer
/external/tensorflow/tensorflow/python/tpu/
Dtpu_optimizer.py100 def compute_gradients(self, loss, var_list=None, **kwargs): member in CrossShardOptimizer
/external/tensorflow/tensorflow/contrib/eager/python/examples/revnet/
Drevnet.py132 def compute_gradients(self, saved_hidden, labels, training=True, l2_reg=True): member in RevNet
/external/tensorflow/tensorflow/contrib/mpi_collectives/
D__init__.py194 def compute_gradients(self, *args, **kwargs): member in DistributedOptimizer
/external/tensorflow/tensorflow/python/training/
Dsync_replicas_optimizer.py208 def compute_gradients(self, *args, **kwargs): member in SyncReplicasOptimizer
Doptimizer.py414 def compute_gradients(self, loss, var_list=None, member in Optimizer
/external/tensorflow/tensorflow/contrib/eager/python/examples/densenet/
Ddensenet_test.py140 def compute_gradients(model, images, labels): function
/external/tensorflow/tensorflow/contrib/eager/python/examples/resnet50/
Dresnet50_test.py53 def compute_gradients(model, images, labels, num_replicas=1): function
/external/tensorflow/tensorflow/contrib/optimizer_v2/
Doptimizer_v2.py726 def compute_gradients(self, member in OptimizerV2