pytorch/torch/ao/nn/quantizable/modules
HDCharles 428cbd7513 [ao] fixing multihead attention convert size (#110407)
Summary: after converting nn.multihead attention we weren't deleting the
old in_proj_weight and in_proj_bias despite not (really) using them.

Test Plan: python test/test_quantization.py -k
"test_custom_module_multi_head_attention"

Reviewers:

Subscribers:

Tasks:

Tags:
Pull Request resolved: https://github.com/pytorch/pytorch/pull/110407
Approved by: https://github.com/jerryzh168
2023-10-03 08:49:12 +00:00
..
__init__.py
activation.py [ao] fixing multihead attention convert size (#110407) 2023-10-03 08:49:12 +00:00
rnn.py [BE]: Update ruff to 0.285 (#107519) 2023-08-22 23:16:38 +00:00