leslie-fang-intel 3a112ebb57 add autocast cpu doc ...

As discussed in https://github.com/pytorch/pytorch/issues/55374#issuecomment-968333614, here we update the cpu autocast operation list in autocast API document.

Pull Request resolved: https://github.com/pytorch/pytorch/pull/68567
Approved by: https://github.com/ezyang

2022-03-22 02:02:43 +00:00

..

amp

add autocast cpu doc

2022-03-22 02:02:43 +00:00

__init__.py

enable torch.cpu.amp.autocast (#57386)

2021-05-20 17:48:36 -07:00