Home
last modified time | relevance | path

Searched defs:get_gradients (Results 1 – 5 of 5) sorted by relevance

/external/tensorflow/tensorflow/python/training/experimental/
Dloss_scale_optimizer_test.py63 def get_gradients(opt, loss, params): function
/external/tensorflow/tensorflow/python/framework/
Dop_callbacks_test.py776 def get_gradients(): function
/external/tensorflow/tensorflow/python/keras/
Doptimizer_v1.py85 def get_gradients(self, loss, params): member in Optimizer
/external/tensorflow/tensorflow/python/keras/optimizer_v2/
Doptimizer_v2.py736 def get_gradients(self, loss, params): member in OptimizerV2
/external/tensorflow/tensorflow/python/keras/mixed_precision/
Dloss_scale_optimizer.py691 def get_gradients(self, loss, params): member in LossScaleOptimizer