pytorch/torch/_subclasses
Edward Z. Yang 2481fc0df4 Add count to FakeTensorMode.__torch_dispatch__ (#93936)
Most calls to fake tensor never hit `FakeTensor.__torch_dispatch__`

Signed-off-by: Edward Z. Yang <ezyang@meta.com>

Pull Request resolved: https://github.com/pytorch/pytorch/pull/93936
Approved by: https://github.com/bdhirsh, https://github.com/albanD
2023-02-03 14:21:11 +00:00
..
__init__.py
fake_tensor.py Add count to FakeTensorMode.__torch_dispatch__ (#93936) 2023-02-03 14:21:11 +00:00
fake_utils.py [MPS] Add group_norm[fwd+backward] and mean_var (take 2) (#91190) 2022-12-22 08:54:37 +00:00
meta_utils.py [Fake Tensor] Allow fake meta by default, delete unused ctor args (#93993) 2023-02-03 09:23:38 +00:00