mirror of https://github.com/llvm/torch-mlir
dfe7513a45
The range of the unsqueeze dim is: [-input.dim() - 1, input.dim() + 1), the bug forgets to add 1. |
||
---|---|---|
.. | ||
TorchToArith | ||
TorchToLinalg | ||
TorchToMhlo | ||
TorchToSCF | ||
TorchToTosa |