Arm backend: Fix bug in ConvertExpandCopyToRepeatPass
In the ConvertExpandCopyToRepeatPass the arguments for the repeat operation are formed incorrectly. For the torch.Tensor.expand operation passing -1 as the size for a dimension means that the size of that dimension does not change. For the DeiT-tiny case, torch.ones(1, 1, 192).expand(1, -1, -1) the pass will prepare arguments to the repeat operation as [1, -1, 1] which will cause an error, in this case the arguments should be [1, 1, 1].
A
Aleksei-grovety committed
b981b2e74f77e099515f81cb86dd2bd8f6cfcb99
Parent: 745d47a
Committed by Fredrik Knutsson <fredrik.knutsson.hunnebo@gmail.com>
on 12/16/2024, 7:48:03 AM