Module: tf.compat.v2.mixed_precision.experimental

Public API for tf.mixed_precision.experimental namespace.

Classes

class DynamicLossScale: Loss scale that dynamically adjusts itself.

class FixedLossScale: Loss scale with a fixed value.

class LossScale: Loss scale base class.

class LossScaleGradientTape: A gradient tape that scales losses and unscales resulting gradients.