Autocast

Context manager or decorator to allow regions of script to run in mixed precision.

Abstract Signature:

Autocast(device_type: str, dtype: dtype, enabled: bool = True, cache_enabled: bool)

PyTorchTensorFlow

PyTorch

API: torch.autocast

Strategy: Direct Mapping

Official Docs ↗

TensorFlow

API: —

Strategy: Macro 'tf.keras.mixed_precision.Policy('mixed_float16')'