guan404ming opened a new pull request, #19863: URL: https://github.com/apache/tvm/pull/19863
## Why The Relax AffineGrid op only handled 2D (4D theta/grid); 5D 3D inputs from ONNX failed. ## How - Generalize struct-info inference to 2D/3D via spatial = size_sinfo->ndim. - Branch TOPI affine_grid compute on 2D vs 3D. - Add the 3D permute path in the frontend and a test_affine_grid_3d case. -- 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]
