Name	Size	Mode	Actions
__pycache__/	-	0755	rm
autocast_mode.py	373	0644	editdlrm
__init__.py	36	0644	editdlrm

import torch class autocast(torch.autocast_mode.autocast): r""" See :class:`torch.autocast`. ``torch.cpu.amp.autocast(args...)`` is equivalent to ``torch.autocast("cpu", args...)`` """ def __init__(self, enabled=True, dtype=torch.bfloat16, cache_enabled=True): super().__init__("cpu", enabled=enabled, dtype=dtype, cache_enabled=cache_enabled)