byteir icon indicating copy to clipboard operation
byteir copied to clipboard

[torch-frontend] fx export FakeTensor weight as SplatElementsAttr.

Open Vremold opened this issue 1 year ago • 0 comments

In the previous PR, we export FakeTensor as a DenseElementsAttr with random values. We change to create a splat DenseElementsAttr in this PR, as FakeTensor doesn't care about tensor values and it saves us a lot of compilation time and memory consumption.

Vremold avatar May 30 '24 16:05 Vremold