pytorch
c112e89c - [quant] Make choose_qparams_optimized return Tensors to preserve dtype (#45530)

Commit
4 years ago
[quant] Make choose_qparams_optimized return Tensors to preserve dtype (#45530) Summary: Pull Request resolved: https://github.com/pytorch/pytorch/pull/45530 Returning double values requires special handling as a return type for aten functions. Instead return tensors where the type is preserved in the tensor dtype Test Plan: python test/test_quantization.py TestQuantizedTensor.test_choose_qparams_optimized Imported from OSS Reviewed By: dskhudia Differential Revision: D24001134 fbshipit-source-id: bec6b17242f4740ab5674be06e0fc30c35eb0379
Author
Parents
Loading