Public API for tf.keras.mixed_precision.experimental namespace.
Classes
class LossScaleOptimizer: An optimizer that applies loss scaling.
class Policy: A dtype policy for a Keras layer.
Functions
get_layer_policy(...): Returns the dtype policy of a layer.
global_policy(...): Returns the global Policy.
set_policy(...): Sets the global Policy.