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
    mindspore.amp.custom_mixed_precision
    mindspore.amp.get_black_list
    mindspore.amp.get_white_list

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

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

    mindspore.amp.all_finite