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