Searched defs:compute_gradients (Results 1 – 5 of 5) sorted by relevance
| /external/tensorflow/tensorflow/python/tpu/ |
| D | tpu_optimizer.py | 111 def compute_gradients(self, loss, var_list=None, **kwargs): member in CrossShardOptimizer
|
| /external/tensorflow/tensorflow/python/training/ |
| D | sync_replicas_optimizer.py | 206 def compute_gradients(self, *args, **kwargs): member in SyncReplicasOptimizer
|
| D | optimizer.py | 423 def compute_gradients(self, loss, var_list=None, member in Optimizer
|
| /external/tensorflow/tensorflow/python/training/experimental/ |
| D | loss_scale_optimizer.py | 84 def compute_gradients(self, member in MixedPrecisionLossScaleOptimizer
|
| /external/tensorflow/tensorflow/python/eager/benchmarks/resnet50/ |
| D | resnet50_test.py | 37 def compute_gradients(model, images, labels, num_replicas=1): function
|