pytorch/torch/_subclasses
Edward Z. Yang f19e07b056 Memoize local_scalar_dense calls, refactor all memos (#125623)
Signed-off-by: Edward Z. Yang <ezyang@meta.com>
Pull Request resolved: https://github.com/pytorch/pytorch/pull/125623
Approved by: https://github.com/eellison
2024-05-11 21:12:35 +00:00
..
__init__.py
fake_impls.py Memoize local_scalar_dense calls, refactor all memos (#125623) 2024-05-11 21:12:35 +00:00
fake_tensor.py Memoize local_scalar_dense calls, refactor all memos (#125623) 2024-05-11 21:12:35 +00:00
fake_utils.py Revert "[fake_impls] Fix seed/offset device for attention kernels (#120839)" (#121447) 2024-03-08 01:48:23 +00:00
functional_tensor.py [RELAND] Switch default behavoir of export IR to be predispatch (#125860) 2024-05-10 17:36:53 +00:00
meta_utils.py functionalize storage resizing, minimal ppFSDP traceable forward (#122434) 2024-05-10 18:09:10 +00:00
schema_check_mode.py Replace follow_imports = silent with normal (#118414) 2024-01-27 02:44:11 +00:00