Skip to content

Add Conv3dInt8 with Fbgemm backend #28480

@xiaomengy

Description

@xiaomengy

🚀 Feature

Add Conv3dInt8 with Fbgemm backend.

Motivation

Currently our quantized conv ops only support 2d conv. In order to support video understanding models, we have to add quantized version of 3d conv. In caffe2 we have the quantized 3d conv implementation with Fbgemm backend. So we can add the quantized Conv3dInt8 with Fbgemm backend here.

Pitch

Alternatives

Additional context

cc @jerryzh168 @jianyuh @dzhulgakov @raghuramank100 @VitalyFedyunin @ngimel @mruberry

Metadata

Metadata

Labels

module: performanceIssues related to performance, either of kernel code or framework glueoncall: 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

No milestone

Relationships

None yet

Development

No branches or pull requests

Issue actions