pytorch
92cfda17 - [PyTorch Edge] Clean up Quantize Tensor code (#66220)

Commit
3 years ago
[PyTorch Edge] Clean up Quantize Tensor code (#66220) Summary: Pull Request resolved: https://github.com/pytorch/pytorch/pull/66220 - Pass pointers rather than tensors to ```quantize_tensor_arm``` to allow for using ```__restrict__``` and to make parallelization easier (as in the next diff on this stack D31205883) - Replace ```auto``` with actual types - Replace raw cast with reinterpret_cast<...> - All of these changes make the code structure similar to that of Dequantize ghstack-source-id: 142166376 Test Plan: same as D31066997 (all tests pass) Reviewed By: kimishpatel Differential Revision: D31444248 fbshipit-source-id: 6a31d090082047263403f415911c199519987595
Author
Parents
Loading