Defined in tensorflow/contrib/mixed_precision/__init__.py
.
Library for mixed precision training.
Classes
class ExponentialUpdateLossScaleManager
: Loss scale manager uses an exponential update strategy.
class FixedLossScaleManager
: Loss scale manager with a fixed loss scale.
class LossScaleManager
: Abstract loss scale manager class.
class LossScaleOptimizer
: An optimizer that applies loss scaling in backprop.