mindspore.amp
================

Loss Scale
----------------

.. autosummary::
    :toctree: amp
    :nosignatures:
    :template: classtemplate.rst

    mindspore.amp.LossScaler
    mindspore.amp.DynamicLossScaler
    mindspore.amp.StaticLossScaler
    mindspore.amp.LossScaleManager
    mindspore.amp.DynamicLossScaleManager
    mindspore.amp.FixedLossScaleManager

Dtype Autocast
--------------------

.. autosummary::
    :toctree: amp
    :nosignatures:
    :template: classtemplate.rst

    mindspore.amp.auto_mixed_precision
    mindspore.amp.build_train_network

Overflow Detection
--------------------

.. autosummary::
    :toctree: amp
    :nosignatures:
    :template: classtemplate.rst

    mindspore.amp.init_status
    mindspore.amp.all_finite