pytorch/test/nn
2025-05-14 15:22:47 +00:00
..
test_convolution.py [CUDA][CUDNN] Dispatch to cuDNN for non-batch-splittable 64-bit NCHW convolutions (#153101) 2025-05-14 15:22:47 +00:00
test_dropout.py Enable additional tests for MPS CI runs (#134356) 2024-10-04 21:52:38 +00:00
test_embedding.py Added Diffentiable per_sample_weights Check to EmbeddingBag.cpp (#142338) 2024-12-11 03:42:17 +00:00
test_init.py Fix unused Python variables in test/nn (#143396) 2024-12-18 03:30:54 +00:00
test_lazy_modules.py Fix broken LazyLinear init (#149693) 2025-03-25 23:49:49 +00:00
test_load_state_dict.py Fix unused Python variables in test/[e-z]* (#136964) 2024-12-18 23:02:30 +00:00
test_module_hooks.py [4/N] Apply py39 ruff and pyupgrade fixes (#143257) 2025-01-04 10:47:51 +00:00
test_multihead_attention.py [ROCm] Update to AOTriton 0.7b (#134498) 2024-09-11 20:34:01 +00:00
test_packed_sequence.py [4/N] Apply py39 ruff and pyupgrade fixes (#143257) 2025-01-04 10:47:51 +00:00
test_parametrization.py Fix unused Python variables in test/nn (#143396) 2024-12-18 03:30:54 +00:00
test_pooling.py add grad_output shape check for adaptive_avg_pool2d_backward (#145241) 2025-03-20 14:10:31 +00:00
test_pruning.py Fix unused Python variables in test/nn (#143396) 2024-12-18 03:30:54 +00:00