ConvolutedDog opened a new pull request, #18478:
URL: https://github.com/apache/tvm/pull/18478
This commit fixes the squeeze operator to behave consistently with PyTorch
by implementing no-op behavior when squeezing dimensions that are not of
size 1.
Previously:
squeeze(x, [1]) on tensor with shape [32, 10, 5] would fail
Now:
squeeze(x, [1]) on tensor with shape [32, 10, 5] returns the original
tensor
without modification, matching PyTorch's behavior
This fixes compatibility issues when converting PyTorch models that use
squeeze with dimensions that may not always be 1 during inference."
Co-authored-by: guan404ming <[email protected]>
--
This is an automated message from the Apache Git Service.
To respond to the message, please log on to GitHub and use the
URL above to go to the specific comment.
To unsubscribe, e-mail: [email protected]
For queries about this service, please contact Infrastructure at:
[email protected]
---------------------------------------------------------------------
To unsubscribe, e-mail: [email protected]
For additional commands, e-mail: [email protected]
