Public API for tf.keras.mixed_precision namespace.
Modules
experimental
module: Public API for tf.keras.mixed_precision.experimental namespace.
Classes
class LossScaleOptimizer
: An optimizer that applies loss scaling to prevent numeric underflow.