pytorch/torch/_dynamo/variables
Wanchao Liang c1db235040 [dynamo] fix module buffers call (#102251)
This PR fixes module buffers call and extract module.buffers similar to
module.parameters

Pull Request resolved: https://github.com/pytorch/pytorch/pull/102251
Approved by: https://github.com/wconstab
2023-05-25 21:26:09 +00:00
..
__init__.py [dynamo] Activation checkpoint higher order ops - Reland 101028 (#101790) 2023-05-18 19:09:14 +00:00
base.py [BE] [2/3] Rewrite super() calls in functorch and torch (#94588) 2023-02-10 21:16:33 +00:00
builder.py Full default dict support in dynamo (#102202) 2023-05-25 01:41:38 +00:00
builtin.py Support unary not on lists (#102210) 2023-05-25 02:45:36 +00:00
constant.py [BE] Enable C419 rule for any all shortcircuiting (#99890) 2023-04-25 15:02:13 +00:00
ctx_manager.py [Dynamo] Fix torch.{cuda/cpu}.amp.autocast arguments binding bug (#101052) 2023-05-10 21:19:18 +00:00
dicts.py Full default dict support in dynamo (#102202) 2023-05-25 01:41:38 +00:00
functions.py [dynamo][higher order op] Support nn.Module calls (#102022) 2023-05-24 21:39:58 +00:00
lists.py Add support for list copy in dynamo export (#100669) 2023-05-05 16:04:19 +00:00
misc.py [dynamo] add itertools.chain support (#102247) 2023-05-25 21:26:09 +00:00
nn_module.py [dynamo] fix module buffers call (#102251) 2023-05-25 21:26:09 +00:00
tensor.py Remove config check in specialize (#102098) 2023-05-24 01:26:22 +00:00
torch.py [dynamo][higher order op] Support nn.Module calls (#102022) 2023-05-24 21:39:58 +00:00
user_defined.py [Dynamo] Improve handling UnspecializedNNModuleVariable side effect (#101141) 2023-05-16 03:57:13 +00:00