Functions 2
get_autocast_cpu_dtype
functionFull reference ↗- tensorplay.amp.autocast_mode.get_autocast_cpu_dtype()[source]
Return the dtype to be used for CPU autocasting.
Warning
Kept for backward compatibility. Prefer
get_autocast_dtype().
get_autocast_gpu_dtype
functionFull reference ↗- tensorplay.amp.autocast_mode.get_autocast_gpu_dtype()[source]
Return the dtype to be used for CUDA autocasting.
Warning
Kept for backward compatibility. Prefer
get_autocast_dtype().
