Skip to content
New issue

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

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

Already on GitHub? # 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!
# for free to join this conversation on GitHub. Already have an account? # to comment
Labels
enhancement New feature or request
Projects
None yet
Development

No branches or pull requests

1 participant