frozenleaves/pytorch
1
0
Fork 0
mirror of https://github.com/pytorch/pytorch.git synced 2025-10-21 05:34:18 +08:00
Code Issues Packages Projects Releases Wiki Activity

Apply UFMT to low traffic torch modules (#106249)

Browse Source 
Signed-off-by: Edward Z. Yang <ezyang@meta.com>

Pull Request resolved: https://github.com/pytorch/pytorch/pull/106249
Approved by: https://github.com/Skylion007
This commit is contained in:
Edward Z. Yang
2023-07-29 10:51:26 -04:00
committed by PyTorch MergeBot
parent a4ebc61f15
commit 3bf922a6ce
163 changed files with 8472 additions and 4412 deletions
Download Patch File Download Diff File Expand all files Collapse all files

2
torch/quantization/fuser_method_mappings.py
View File

@ -7,9 +7,9 @@ If you are adding a new entry/functionality, please, add it to the
here.
"""
from torch.ao.quantization.fuser_method_mappings import (
_DEFAULT_OP_LIST_TO_FUSER_METHOD,
fuse_conv_bn,
fuse_conv_bn_relu,
fuse_linear_bn,
_DEFAULT_OP_LIST_TO_FUSER_METHOD,
get_fuser_method,
)