Xia, Weiwen 1a722f62c2 [Quant][X86] add an op to compute uint8 batch norm 2d (#152811) ... **Summary** This PR adds a new op, `onednn.qbatch_norm2d`, which accepts uint8 inputs on CPU device (instead of QuantizedCPU). The new ops are implemented with AVX512 instructions and it provides similar performance as its counterpart for QuantizedCPU device `quantized.batch_norm2d`. The new op supports output dtypes other than uint8 (fp32, fp16 and bf16 are supported). **Test plan** ``` pytest test/quantization/core/test_quantized_op.py -k test_int8_batch_norm_onednn ``` Pull Request resolved: https://github.com/pytorch/pytorch/pull/152811 Approved by: https://github.com/leslie-fang-intel, https://github.com/jerryzh168, https://github.com/jgong5 ghstack dependencies: #152411		2025-05-16 06:13:40 +00:00
..
experimental	add torch.float4_e2m1fn_x2 to PyTorch (#148791)	2025-03-27 17:32:20 +00:00
__init__.py
test_backend_config.py	Fix unused Python variables in test/[e-z]* (#136964)	2024-12-18 23:02:30 +00:00
test_docs.py	Revert "Use absolute path path.resolve() -> path.absolute() (#129409)"	2025-01-04 14:17:20 +00:00
test_quantized_functional.py	Fix broken URLs (#152237)	2025-04-27 09:56:42 +00:00
test_quantized_module.py	Fix broken URLs (#152237)	2025-04-27 09:56:42 +00:00
test_quantized_op.py	[Quant][X86] add an op to compute uint8 batch norm 2d (#152811)	2025-05-16 06:13:40 +00:00
test_quantized_tensor.py	Fix unused Python variables in test/[e-z]* (#136964)	2024-12-18 23:02:30 +00:00
test_top_level_apis.py
test_utils.py	[reland] Add int1 to int7 dtypes (#137928)	2024-10-18 02:02:08 +00:00
test_workflow_module.py	Fix unused Python variables in test/[e-z]* (#136964)	2024-12-18 23:02:30 +00:00
test_workflow_ops.py	PEP585 update - test (#145176)	2025-01-22 04:48:28 +00:00