Elias Ellison c1d11f40c9 [FakeTensor] Use the device of the meta tensor for fallback kernel (#80193) ... Otherwise, I would run into issues such as fp16 not supported for Conv Cpu, or cudnn_rnn not implemented for cpu. Pull Request resolved: https://github.com/pytorch/pytorch/pull/80193 Approved by: https://github.com/Chillee		2022-06-24 20:00:07 +00:00
..
__init__.py	Add option for allowing non-fake inputs, add deepcopy impl	2022-06-17 19:36:26 +00:00
fake_tensor.py	[FakeTensor] Use the device of the meta tensor for fallback kernel (#80193)	2022-06-24 20:00:07 +00:00
meta_utils.py	Replace TensorMeta with FakeTensor	2022-06-05 11:51:27 +00:00