mikeseven opened a new pull request, #14930: URL: https://github.com/apache/tvm/pull/14930
relay.frontend.from_pytorch maps torch.squeeze() using 1 axis while pytorch allows multiple axis. This fixes that. Note that this pattern seems to be used for other functions. This frontend may need to be updated elsewhere. I only tested for squeeze for my current needs. -- 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]
