mindspore.amp.get_white_list
- mindspore.amp.get_white_list()[源代码]
提供用于自动混合精度的内置白名单的拷贝。
当前的内置白名单内容为：
[
mindspore.nn.Conv1d,mindspore.nn.Conv2d,mindspore.nn.Conv3d,mindspore.nn.Conv1dTranspose,mindspore.nn.Conv2dTranspose,mindspore.nn.Conv3dTranspose,mindspore.nn.Dense,mindspore.nn.LSTMCell,mindspore.nn.RNNCell,mindspore.nn.GRUCell,mindspore.ops.Conv2D,mindspore.ops.Conv3D,mindspore.ops.Conv2DTranspose,mindspore.ops.Conv3DTranspose,mindspore.ops.MatMul,mindspore.ops.BatchMatMul,mindspore.ops.PReLU,mindspore.ops.ReLU,mindspore.ops.Ger]- 返回：
list：内置白名单的拷贝。
样例：
>>> from mindspore import amp >>> white_list = amp.get_white_list() >>> print(white_list) [<class 'mindspore.nn.layer.conv.Conv1d'>, <class 'mindspore.nn.layer.conv.Conv2d'>, <class 'mindspore.nn.layer.conv.Conv3d'>, <class 'mindspore.nn.layer.conv.Conv1dTranspose'>, <class 'mindspore.nn.layer.conv.Conv2dTranspose'>, <class 'mindspore.nn.layer.conv.Conv3dTranspose'>, <class 'mindspore.nn.layer.basic.Dense'>, <class 'mindspore.nn.layer.rnn_cells.LSTMCell'>, <class 'mindspore.nn.layer.rnn_cells.RNNCell'>, <class 'mindspore.nn.layer.rnn_cells.GRUCell'>, <class 'mindspore.ops.operations.nn_ops.Conv2D'>, <class 'mindspore.ops.operations.nn_ops.Conv3D'>, <class 'mindspore.ops.operations.nn_ops.Conv2DTranspose'>, <class 'mindspore.ops.operations.nn_ops.Conv3DTranspose'>, <class 'mindspore.ops.operations.nn_ops.Conv2DBackpropInput'>, <class 'mindspore.ops.operations.math_ops.MatMul'>, <class 'mindspore.ops.operations.math_ops.BatchMatMul'>, <class 'mindspore.ops.operations.nn_ops.PReLU'>, <class 'mindspore.ops.operations.nn_ops.ReLU'>, <class 'mindspore.ops.operations.math_ops.Ger'>]