pytorch/torch/quantization
Jerry Zhang c627211651 [quant][fx][graphmode][be] Change the type for output of convert to be torch.nn.Module (#69959)
Summary:
Pull Request resolved: https://github.com/pytorch/pytorch/pull/69959

GraphModule is an implementation detail, We don't want to expose it in quantization apis

Test Plan:
python test/test_quantization.py TestQuantizeFx.test_quantized_model_type

Imported from OSS

Reviewed By: supriyar

Differential Revision: D33119103

fbshipit-source-id: d8736ff08b42ee009d6cfd74dcb3f6150f71f3d2
2021-12-29 20:33:32 -08:00
..
fx fx quant: move _parent_name to common utils (#69720) 2021-12-17 05:59:46 -08:00
__init__.py torch.ao migration: fake_quantize.py, phase 1 (#64814) 2021-09-13 15:22:28 -07:00
_numeric_suite_fx.py torch.ao migration: numeric suite, eager and fx (#64817) 2021-09-12 12:00:45 -07:00
_numeric_suite.py torch.ao migration: numeric suite, eager and fx (#64817) 2021-09-12 12:00:45 -07:00
fake_quantize.py torch.ao migration: fake_quantize.py, phase 1 (#64814) 2021-09-13 15:22:28 -07:00
fuse_modules.py [quant] AO migration of the fuse_modules.py (phase 1) (#64913) 2021-09-15 17:28:47 -07:00
fuser_method_mappings.py [quant]ao migration for quantization mappings and fuser method mappings hg mv (#64985) 2021-09-16 12:59:20 -07:00
observer.py [quant] ao migration of observer and qconfig (#64982) 2021-09-16 10:33:16 -07:00
qconfig.py [quant] ao migration of observer and qconfig (#64982) 2021-09-16 10:33:16 -07:00
quant_type.py [quant] AO migration of the quant_types.py (phase 1) (#64916) 2021-09-15 17:30:00 -07:00
quantization_mappings.py [quant]ao migration for quantization mappings and fuser method mappings hg mv (#64985) 2021-09-16 12:59:20 -07:00
quantize_fx.py [quant][fx][graphmode][be] Change the type for output of convert to be torch.nn.Module (#69959) 2021-12-29 20:33:32 -08:00
quantize_jit.py torch.ao migration: quantize_jit.py phase1 (#64860) 2021-09-13 08:41:48 -07:00
quantize.py [quant] Removing unnecessary import from torch/quantization/quantize.py (#64910) 2021-09-15 09:39:04 -07:00
stubs.py torch.ao migration: stubs.py phase 1 (#64861) 2021-09-13 08:40:29 -07:00
utils.py [quant] AO migration of the torch/quantization/utils.py (phase 1) (#64919) 2021-09-16 21:30:18 -07:00