pytorch/torch/_subclasses
Pian Pawakapan 4be831ba2d [draft_export] fix dense-in-memory check for inferring fakes (#145653)
Test Plan: fixes check for dense tensors with size-1 dimensions

Differential Revision: D68644028

Pull Request resolved: https://github.com/pytorch/pytorch/pull/145653
Approved by: https://github.com/zou3519
2025-01-28 02:52:14 +00:00
..
__init__.py
_fake_tensor_utils.py PEP585 update - torch/_higher_order_ops torch/_subclasses torch/backends torch/compiler torch/cuda torch/masked torch/mtia torch/nested (#145202) 2025-01-20 22:37:26 +00:00
fake_impls.py Remove FFT from stride incorrect ops (#145080) 2025-01-27 04:26:04 +00:00
fake_tensor.py [draft_export] fix dense-in-memory check for inferring fakes (#145653) 2025-01-28 02:52:14 +00:00
fake_utils.py PEP585 update - torch/_higher_order_ops torch/_subclasses torch/backends torch/compiler torch/cuda torch/masked torch/mtia torch/nested (#145202) 2025-01-20 22:37:26 +00:00
functional_tensor.py PEP585 update - torch/_higher_order_ops torch/_subclasses torch/backends torch/compiler torch/cuda torch/masked torch/mtia torch/nested (#145202) 2025-01-20 22:37:26 +00:00
meta_utils.py Revert "pickler for GraphModule (#141659)" 2025-01-27 22:39:30 +00:00
schema_check_mode.py