vacu9708 opened a new pull request, #18090: URL: https://github.com/apache/tvm/pull/18090
# Summary This PR follows up on https://github.com/apache/tvm/pull/18072 by updating the Reduce operators to accept axes as an input instead of an attribute. # Changes - Support axes as an input for Reduce ops (e.g., ReduceL2, ReduceMax, …) - Add corresponding test cases | Axes provided? | noop_with_empty_axes | Behavior | |:-----:|:-----:|:----------------------------------------------:| | No | 0 | Reduce over **all** dimensions | | No | 1 | Return the input **unchanged** | | Yes | any | Reduce over the **specified** axes | -- 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]
