Skip to content

Support Standalone Batch Norm #11586

@mcr229

Description

@mcr229

🚀 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

Metadata

Metadata

Assignees

Labels

Type

No type

Projects

Status

No status

Milestone

No milestone

Relationships

None yet

Development

No branches or pull requests

Issue actions