openvinotoolkit/nncf

[TorchFX] Use FakeTensors in meta instead of actual tensors

daniil-lyakhov opened this issue ยท 0 comments

๐Ÿš€ 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!