Announcing the TensorFlow Dev Summit 2020 Learn more

Module: tf.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.