Home
last modified time | relevance | path

Searched refs:strategy_supports_loss_scaling (Results 1 – 3 of 3) sorted by relevance

/external/tensorflow/tensorflow/python/keras/mixed_precision/
Dloss_scale_optimizer.py797 if not strategy_supports_loss_scaling():
1170 def strategy_supports_loss_scaling(): function
/external/tensorflow/tensorflow/python/keras/engine/
Dbase_layer_v1.py1758 not loss_scale_optimizer.strategy_supports_loss_scaling()):
Dbase_layer.py2356 not loss_scale_optimizer.strategy_supports_loss_scaling()):