pytorch
6114b0f9 - [quant][core][improvement][feature] Enabled support for quantized fill of nhwc tensors

Commit
2 years ago
[quant][core][improvement][feature] Enabled support for quantized fill of nhwc tensors Summary: Previously, filling a quantized tensor only worked for nchw tensors. This PR enables support for nhwc tensors. Test cases were added for per tensor and per channel quantized NHWC tensors. Test Plan: ``` python test/test_quantization.py -k test_qtensor_fill_per_channel_nhwc python test/test_quantization.py -k test_qtensor_fill_per_tensor_nhwc ``` Pull Request resolved: https://github.com/pytorch/pytorch/pull/79025 Approved by: https://github.com/jerryzh168
Author
Committer
Parents
Loading