mindspore.device_context.gpu.op_precision.conv_allow_tf32

View Source On Gitee
mindspore.device_context.gpu.op_precision.conv_allow_tf32(value)[source]

Whether to convert FP32 to TF32 for Conv operators. For detailed information, please refer to CUBLAS_COMPUTE_32F_FAST_TF32.

Parameters

value (bool) – Whether to convert FP32 to HF32 for Conv operators. If not configured, the framework defaults to True.

Examples

>>> import mindspore as ms
>>> ms.device_context.gpu.op_precision.conv_allow_tf32(False)