pytorch/test/nn
2025-05-20 20:19:03 +00:00
..
test_convolution.py [CUDA][CUDNN] Dispatch to cuDNN for non-batch-splittable 64-bit NCHW convolutions (#153101) 2025-05-20 20:19:03 +00:00
test_dropout.py
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
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