Website Blog TensorPlay documentation

Automatic differentiation package - tensorplay.autograd Automatic Mixed Precision package - tensorplay.amp Distributed communication package - tensorplay.distributed Multiprocessing package - tensorplay.multiprocessing Quantization tensorplay tensorplay.audio tensorplay.compiler tensorplay.cuda tensorplay.fft tensorplay.futures tensorplay.hub tensorplay.library tensorplay.linalg tensorplay.nn tensorplay.nn.functional tensorplay.nn.init tensorplay.optim tensorplay.random tensorplay.sparse tensorplay.special tensorplay.stax tensorplay.utils.checkpoint tensorplay.utils.data tensorplay.vision Upstream anchor labels

autocast GradScaler tensorplay.amp.autocast_mode.custom_bwd tensorplay.amp.autocast_mode.custom_fwd tensorplay.amp.autocast_mode.get_autocast_cpu_dtype tensorplay.amp.autocast_mode.get_autocast_gpu_dtype tensorplay.amp.autocast_mode.is_autocast_available tensorplay.amp.is_autocast_available

API symbolsamp

Copy

View Markdown Download .md

tensorplay.amp.autocast_mode.get_autocast_gpu_dtype

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().

Help improve this page

Found an error, an unclear step, or a missing example?

Was this page helpful?

Suggest an edit Report an issue

tensorplay.amp.autocast_mode.get_autocast_cpu_dtype

Return the dtype to be used for CPU autocasting.

tensorplay.amp.autocast_mode.is_autocast_available

Next Page