byteir
byteir copied to clipboard
[torch-frontend] fx export FakeTensor weight as SplatElementsAttr.
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.