Skip to content
New issue

Have a question about this project? Sign up for a free GitHub account to open an issue and contact its maintainers and the community.

By clicking “Sign up for GitHub”, you agree to our terms of service and privacy statement. We’ll occasionally send you account related emails.

Already on GitHub? Sign in to your account

[TorchFX] Deptwise convolution support #2896

Merged

Conversation

daniil-lyakhov
Copy link
Collaborator

Changes

Depthwise convolution subtypes are integrated to torchFX nncf graph builder

Reason for changes

To correctly quantize deptiwise convolutions

Related tickets

#2766

Tests

tests/torch/fx/test_models.py is updated

nncf/experimental/torch/fx/nncf_graph_builder.py Outdated Show resolved Hide resolved
nncf/experimental/torch/fx/nncf_graph_builder.py Outdated Show resolved Hide resolved
nncf/experimental/torch/fx/nncf_graph_builder.py Outdated Show resolved Hide resolved
tests/torch/fx/test_models.py Outdated Show resolved Hide resolved
@AlexanderDokuchaev AlexanderDokuchaev merged commit c57df43 into openvinotoolkit:develop Aug 23, 2024
13 checks passed
Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment
Labels
experimental NNCF PT Pull requests that updates NNCF PyTorch
Projects
None yet
Development

Successfully merging this pull request may close these issues.

2 participants