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

[ao] fuser_method_mappings.py fixing public v private (#87516)

Browse Source 
Summary: made _get_valid_patterns, _DEFAULT_PATTERN_TO_FUSER_METHOD,
_reverse3, _reverse2, _reverse_sequential_wrapper2,
_DEFAULT_OP_LIST_TO_FUSER_METHOD, _sequential_wrapper2 private

Test Plan: python test/test_public_bindings.py

Reviewers:

Subscribers:

Tasks:

Tags:

Differential Revision: [D40709281](https://our.internmc.facebook.com/intern/diff/D40709281)

Pull Request resolved: https://github.com/pytorch/pytorch/pull/87516
Approved by: https://github.com/jcaip
This commit is contained in:
HDCharles
2022-11-08 07:59:11 -08:00
committed by PyTorch MergeBot
parent 37b468ac77
commit ad2eba802c
9 changed files with 57 additions and 68 deletions
Download Patch File Download Diff File Expand all files Collapse all files

2
torch/quantization/fuser_method_mappings.py
View File

@ -10,6 +10,6 @@ from torch.ao.quantization.fuser_method_mappings import (
fuse_conv_bn,
fuse_conv_bn_relu,
fuse_linear_bn,
DEFAULT_OP_LIST_TO_FUSER_METHOD,
_DEFAULT_OP_LIST_TO_FUSER_METHOD,
get_fuser_method,
)