Module: tf.keras.mixed_precision.experimental
Mixed precision API.
Aliases:
- Module
tf.compat.v2.keras.mixed_precision.experimental
Classes
class LossScaleOptimizer
: An optimizer that applies loss scaling.
class Policy
: A dtype policy for a Keras layer.
Functions
global_policy(...)
: Returns the global Policy.
set_policy(...)
: Sets the global Policy.