Preserve memory format in qconv op (#49533)
Summary:
* qconv used to return NHWC no matter the input format
* this change returns NCHW format if the input was NCHW
Pull Request resolved: https://github.com/pytorch/pytorch/pull/49533
Test Plan:
pytest test/quantization/test_quantized_op.py::\
TestQuantizedConv::test_qconv2d_preserve_mem_format
Fixes https://github.com/pytorch/pytorch/issues/47295
Reviewed By: kimishpatel
Differential Revision: D25609205
Pulled By: axitkhurana
fbshipit-source-id: 83f8ca4a1496a8a4612fc3da082d727ead257ce7