Skip to content

Add support for fusing ConvBnReLU3d  #50002

@dmenig

Description

@dmenig

🚀 Feature

Add support for fusion of quantized 3d layers.

Motivation

I wanna be able to do quantization for 3d models such as 3d resnets but the support is lacking for fusing 3d layers in pytorch quantization modules.

Pitch

I believe what is missing is the definition of the ConvBnReLU3d class.

cc @jerryzh168 @jianyuh @dzhulgakov @raghuramank100 @jamesr66a @vkuzo

Metadata

Metadata

Assignees

Labels

oncall: quantizationQuantization support in PyTorchtriagedThis issue has been looked at a team member, and triaged and prioritized into an appropriate module

Type

No type

Projects

No projects

Milestone

Relationships

None yet

Development

No branches or pull requests

Issue actions