OSSForks/pytorch
1
0
Fork 0
mirror of https://github.com/zebrajr/pytorch.git synced 2025-12-06 00:20:18 +01:00
Code Issues Actions 38 Packages Projects Releases Wiki Activity

[quant][graphmode] Rename _quantize_script.py to quantize_script.py (#39122)

Browse Source 
Summary: Pull Request resolved: https://github.com/pytorch/pytorch/pull/39122

Test Plan: Imported from OSS

Differential Revision: D21757619

fbshipit-source-id: 603c020aaaf6f467e63f15b4f271fe946d9fb949
This commit is contained in:
Jerry Zhang 2020-05-29 12:24:34 -07:00 committed by Facebook GitHub Bot
parent a50d781c03
commit 1d0ec50a02
5 changed files with 11 additions and 10 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

2
mypy.ini
View File

@ -67,7 +67,7 @@ ignore_errors = True
[mypy-torch.quantization.fake_quantize]
ignore_errors = True
[mypy-torch.quantization._quantize_script]
[mypy-torch.quantization.quantize_script]
ignore_errors = True
[mypy-torch.quantization._numeric_suite]

2
test/quantization/test_quantize.py
View File

@ -29,7 +29,7 @@ from torch.quantization import (
default_histogram_observer,
)
from torch.quantization._quantize_script import (
from torch.quantization.quantize_script import (
quantize_script,
quantize_dynamic_script
)

16
test/quantization/test_quantize_script.py
View File

@ -14,14 +14,14 @@ from torch.quantization import default_per_channel_weight_observer
from torch.quantization import default_qconfig
from torch.quantization import get_default_qconfig
# torch.quantization._quantize_script
from torch.quantization._quantize_script import script_qconfig
from torch.quantization._quantize_script import prepare_script
from torch.quantization._quantize_script import convert_script
from torch.quantization._quantize_script import quantize_script
from torch.quantization._quantize_script import prepare_dynamic_script
from torch.quantization._quantize_script import convert_dynamic_script
from torch.quantization._quantize_script import quantize_dynamic_script
# torch.quantization.quantize_script
from torch.quantization.quantize_script import script_qconfig
from torch.quantization.quantize_script import prepare_script
from torch.quantization.quantize_script import convert_script
from torch.quantization.quantize_script import quantize_script
from torch.quantization.quantize_script import prepare_dynamic_script
from torch.quantization.quantize_script import convert_dynamic_script
from torch.quantization.quantize_script import quantize_dynamic_script
# Testing utils
from torch.testing._internal.common_quantization import test_only_eval_fn as _test_only_eval_fn

1
torch/quantization/__init__.py
View File

@ -5,6 +5,7 @@ from .qconfig import *
from .fake_quantize import *
from .fuse_modules import fuse_modules
from .stubs import *
from .quantize_script import *
def default_eval_fn(model, calib_data):
r"""

0
torch/quantization/_quantize_script.py → torch/quantization/quantize_script.py
View File