merrymercy opened a new pull request #7163: URL: https://github.com/apache/tvm/pull/7163
There is an exponent operation in softmax. We can use `fast_exp` from #4790 to accelerate it. This pr adds a new op `topi.nn.fast_softmax` to use this optimization. This op is intended to be used by auto-scheduler. I don't have a manual schedule for it. Due to the lacking of manual schedule, I don't add it to relay's FastMath pass, because other people might use FastMath pass without auto-scheduler. ---------------------------------------------------------------- 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. For queries about this service, please contact Infrastructure at: [email protected]
