Home
last modified time | relevance | path

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

/aosp_15_r20/external/tensorflow/tensorflow/python/training/experimental/
H A Dloss_scale_optimizer_test.py58 def get_gradients(opt, loss, params): function
/aosp_15_r20/external/tensorflow/tensorflow/python/framework/
H A Dop_callbacks_test.py772 def get_gradients(): function
/aosp_15_r20/external/tensorflow/tensorflow/python/keras/
H A Doptimizer_v1.py80 def get_gradients(self, loss, params): member in Optimizer
/aosp_15_r20/external/tensorflow/tensorflow/python/keras/mixed_precision/
H A Dloss_scale_optimizer.py583 def get_gradients(self, loss, params): member in LossScaleOptimizer
/aosp_15_r20/external/tensorflow/tensorflow/python/keras/optimizer_v2/
H A Doptimizer_v2.py742 def get_gradients(self, loss, params): member in OptimizerV2