Support Standalone Batch Norm
Open
@ayushrajdahal is already working on this.
Since Jun 16, 2025.
good first issue
- Dominant language
- Python
- Stars
- 5k
- Forks
- 1.2k
- Avg merge
- 2d 10h
- Merged PRs (30d)
- 581
Description
🚀 The feature, motivation and pitch
Currently XNNPACK doesn't support Batch Norm, however, we can generally support it by fusing it with convolution before hand. However for standalone batch norms we can also convert them into depthwise conv1ds. Take a look at the following code here:
https://github.com/pytorch/executorch/blob/main/backends/xnnpack/_passes/fuse_batch_norm_with_conv.py
Alternatives
No response
Additional context
No response
RFC (Optional)
No response
Contributor guide
First steps
- Read the whole issue, then the project's contributing guide.
- Comment on the issue to say you are picking it up — it saves two people doing the same work.
- Fork the repository and make your change on a branch.
- Open a pull request that references the issue number.
Assessment
This issue has not been assessed yet.