KAI ZHAO 1aa14fcb14 Fix the "tensors to be on the same device" error in HistogramObserver (#59234) ...

Summary:
Fixes https://github.com/pytorch/pytorch/issues/59075

This PR fixes the "tensors to be on the same device" error in `HistogramObserver`.

Pull Request resolved: https://github.com/pytorch/pytorch/pull/59234

Reviewed By: jbschlosser

Differential Revision: D28837572

Pulled By: vkuzo

fbshipit-source-id: ff7c3229ced7de2cdd8f76d526f0fd33ac643216

2021-06-03 13:30:56 -07:00

..

fx

[quant][graphmode][fx][refactor] Split quantize.py to prepare.py and convert.py (#59353)

2021-06-02 23:52:39 -07:00

ns

[quant][graphmode][fx][refactor] Split quantize.py to prepare.py and convert.py (#59353)

2021-06-02 23:52:39 -07:00

__init__.py

Un-ignore F403 in .flake8 (#55838)

2021-04-13 09:24:07 -07:00

_correct_bias.py

Remove py2 compatible future imports (#44735)

2020-09-16 12:55:57 -07:00

_equalize.py

[quant] Eager mode equalization support for ConvReLU and LinearReLU (#58792)

2021-05-24 17:25:13 -07:00

_learnable_fake_quantize.py

[quant][graphmode][fx][refactor] Split quantize.py to prepare.py and convert.py (#59353)

2021-06-02 23:52:39 -07:00

_numeric_suite_fx.py

ns for fx: move relatedness mapping to mappings file (#57171)

2021-05-05 06:29:11 -07:00

_numeric_suite.py

Add lint for unqualified type: ignore (#56290)

2021-04-21 08:07:23 -07:00

fake_quantize.py

memory efficient per-channel fq: use it everywhere, delete old version (#51265)

2021-01-28 19:42:25 -08:00

fuse_modules.py

Add lint for unqualified noqa (#56272)

2021-04-19 13:16:18 -07:00

fuser_method_mappings.py

fix docstring for fusing functions (#58638)

2021-05-24 18:27:22 -07:00

observer.py

Fix the "tensors to be on the same device" error in HistogramObserver (#59234)

2021-06-03 13:30:56 -07:00

qconfig.py

Un-ignore F403 in .flake8 (#55838)

2021-04-13 09:24:07 -07:00

quant_type.py

[quant][graphmode][fx] custom_module support static/dynamic/weight_only quant (#46786)

2020-10-27 21:41:33 -07:00

quantization_mappings.py

fix nn.MHA scriptability (#58727)

2021-05-26 15:29:49 -07:00

quantize_fx.py

[quant][fx] Validate qconfig_dict keys (#58566)

2021-05-21 15:20:05 -07:00

quantize_jit.py

Enable the quantization on XPU devices (#54857)

2021-05-20 17:02:13 -07:00

quantize.py

Back out "[quant][graphmode][fx] Separate handling Copy operator to a helper function" (#55388)

2021-04-06 14:20:36 -07:00

stubs.py

type check for torch.quantization.stubs (#46475)

2020-10-16 15:34:23 -07:00

utils.py

[quant][graphmode][fx] Fix a condition check for CopyNode (#53585)

2021-03-11 09:32:20 -08:00