initial_loss_scale (definition) in projects: aosp_15_r20

Searched defs:initial_loss_scale (Results 1 – 3 of 3) sorted by relevance

/aosp_15_r20/external/tensorflow/tensorflow/python/training/experimental/
H A D	loss_scale.py	`319 initial_loss_scale=2 ** 15, # See docstring for why this is big. argument 351 def initial_loss_scale(self): member in DynamicLossScale`
H A D	loss_scale_test.py	`123 initial_loss_scale=1., argument`
/aosp_15_r20/external/tensorflow/tensorflow/python/keras/mixed_precision/
H A D	loss_scale_optimizer.py	`100 initial_loss_scale, argument 187 def initial_loss_scale(self): member in _DynamicLossScaleState`