balaram-cadence opened a new pull request, #13846: URL: https://github.com/apache/tvm/pull/13846
Currently, ONNX frontend imports Unsqueeze as a sequence of expand_dims in Relay. This works fine when the axes are positive. However, for negative axes we need to compute the output rank and add the negative axis value to get its equivalent positive axis value. This commit computes the output rank as the sum of the input rank of data and the number of axis in the axes array and convert negative axes to positive. This commit also adds testcases for unsqueeze with negative axes and verifies the result with onnx runtime. -- 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]
