pytorch/test/quantization
Jerry Zhang 1719cb82f3 [quant][graphmode][fx] Support preserving attributes in deepcopy of observed/quantized graphmodule (#56550)
Summary:
Pull Request resolved: https://github.com/pytorch/pytorch/pull/56550

Add support for preserving a list of attributes on observed/quantized GraphModule

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

Imported from OSS

Reviewed By: vkuzo, kazhang

Differential Revision: D27899317

fbshipit-source-id: ebf21334715e5ab764aaa27eed534cc0cdf9f2b5
2021-04-22 15:02:44 -07:00
..
serialized
__init__.py
test_backward_compatibility.py
test_bias_correction.py Numeric Suite: Swap with shadow modules only for quantized part of model (#51052) 2021-02-04 11:40:30 -08:00
test_equalize.py
test_fusion_passes.py
test_numeric_suite.py Numeric Suite: Swap with shadow modules only for quantized part of model (#51052) 2021-02-04 11:40:30 -08:00
test_numeric_suite_fx.py [quant][graphmode][fx] Produce torch.cat instead of torch.ops.quantized.cat (#54924) 2021-04-21 10:58:09 -07:00
test_qat_module.py Remove legacy constructor calls from pytorch codebase. (#54142) 2021-04-11 15:45:17 -07:00
test_quantize.py quant: fix conv transpose with qconfig == None (#52844) 2021-02-25 11:52:30 -08:00
test_quantize_fx.py [quant][graphmode][fx] Support preserving attributes in deepcopy of observed/quantized graphmodule (#56550) 2021-04-22 15:02:44 -07:00
test_quantize_jit.py Add padding_idx argument to EmbeddingBag (#49237) 2021-04-14 09:38:01 -07:00
test_quantized_functional.py [reland][quant] Remove nn.quantized.ReLU module and nn.quantized.functional.relu (#47415) (#48038) 2020-11-17 09:52:21 -08:00
test_quantized_module.py [quantization] Fix deepcopy on quantized ConvNd (#56154) 2021-04-15 15:18:22 -07:00
test_quantized_op.py Fix overflow issue in quantized instance_norm/layer_norm/group_norm (#54872) 2021-04-13 10:21:38 -07:00
test_quantized_tensor.py
test_workflow_module.py Remove legacy constructor calls from pytorch codebase. (#54142) 2021-04-11 15:45:17 -07:00