ZephyrLi-pro opened a new pull request, #19909: URL: https://github.com/apache/tvm/pull/19909
This PR extends `rfactor` reducer pattern matching to support TOPI `argmax`/`argmin` with `select_last_index=True`. For `select_last_index=True`, ties should be resolved by choosing the larger index. The added reducer patterns preserve the existing value comparison semantics, then use `lhs_idx > rhs_idx` as the tie-break condition. Tests are added for applying `rfactor` to `topi.argmax` and `topi.argmin` with `select_last_index=True`. -- 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]
