Skip to content

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

Open
@daniil-lyakhov

Description

@daniil-lyakhov

🚀 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!

Metadata

Metadata

Labels

enhancementNew feature or request

Type

No type

Projects

No projects

Milestone

No milestone

Relationships

None yet

Development

No branches or pull requests

Issue actions