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] Use FakeTensors in meta instead of actual tensors #2934

Open
1 task
daniil-lyakhov opened this issue Aug 30, 2024 · 0 comments
Open
1 task

[TorchFX] Use FakeTensors in meta instead of actual tensors #2934

daniil-lyakhov opened this issue Aug 30, 2024 · 0 comments
Assignees
Labels
enhancement New feature or request

Comments

@daniil-lyakhov
Copy link
Collaborator

daniil-lyakhov commented Aug 30, 2024

🚀 Feature request

Current implementation is setting actual tensors as meta value (An exapmle: https://github.com/openvinotoolkit/nncf/blob/develop/nncf/experimental/torch/fx/transformations.py#L30-L46). The task is to optimize this process by creating fake tensors instead. Other problem - not every constant in captured graphs has meta val. The solution should cover such constants as well

Feature Use Case

No response

Are you going to submit a PR?

  • Yes I'd like to help by submitting a PR!
Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment
Labels
enhancement New feature or request
Projects
None yet
Development

No branches or pull requests

1 participant