SetAutocastDtype

Sets the autocast dtype.

Abstract Signature:

SetAutocastDtype(device_type: str, dtype: DType)

PyTorch

API: torch.set_autocast_dtype
Strategy: Direct Mapping
Official Docs ↗

Keras

API: keras.mixed_precision.set_dtype_policy
Strategy: Direct Mapping
Official Docs ↗