pytorch/torch/_subclasses
kshitij12345 d552c271db [pt2] grad support (#102264)
Teach dynamo about grad

Pull Request resolved: https://github.com/pytorch/pytorch/pull/102264
Approved by: https://github.com/zou3519
2023-06-21 10:13:09 +00:00
..
__init__.py
fake_tensor.py Replace _prims_common.check with torch._check* (#103240) 2023-06-21 00:46:17 +00:00
fake_utils.py [operator_compile_check] Add FakeTensor testing (#103595) 2023-06-16 16:55:51 +00:00
meta_utils.py [pt2] grad support (#102264) 2023-06-21 10:13:09 +00:00
schema_check_mode.py Get SchemaCheckMode to error on ops that return inputs directly. Expose as a dynamo backend, eager_debug (#99744) 2023-04-27 20:12:42 +00:00