pytorch/test/nn
FFFrog bc3f0d341a LazyBatchNorm{1-3}d support dict&set (#109015)
Fixes #105292

As the title shown ,LazyBatchNorm don`t support dict&set, keep consistent with BatchNorm{1-3}d.
Pull Request resolved: https://github.com/pytorch/pytorch/pull/109015
Approved by: https://github.com/mikaylagawarecki
2023-09-12 09:09:59 +00:00
..
test_convolution.py [BE] Enable ruff's UP rules and autoformat nn/ mps/ and torch/ (#105436) 2023-07-21 07:38:46 +00:00
test_dropout.py [test_nn] add custom device support for dropout tests、lazy_modules te… (#106609) 2023-08-11 09:14:34 +00:00
test_embedding.py Lift user defined attributes into inputs for certain cases (user defined types and tensors) (#103386) 2023-06-20 23:45:19 +00:00
test_init.py
test_lazy_modules.py LazyBatchNorm{1-3}d support dict&set (#109015) 2023-09-12 09:09:59 +00:00
test_module_hooks.py [BE] Enable ruff's UP rules and autoformat nn/ mps/ and torch/ (#105436) 2023-07-21 07:38:46 +00:00
test_multihead_attention.py [BE]: Update ruff to 0.285 (#107519) 2023-08-22 23:16:38 +00:00
test_packed_sequence.py [BE]: Apply ruff PERF fixes to torch (#104917) 2023-07-11 20:45:21 +00:00
test_parametrization.py Back out "Reland "Make adding buffers more like adding parameters (#104069)" (#106224)" (#106743) 2023-08-08 15:27:34 +00:00
test_pooling.py add Half support for maxpool on CPU (#98819) 2023-09-05 18:23:41 +00:00
test_pruning.py