Ubospica opened a new pull request, #14060: URL: https://github.com/apache/tvm/pull/14060
This PR supports the cases when input does not contain batches for `topi.nll_loss`. When there is no batches, the shape of the prediction parameter is `(C,)`, the shape of the target parameter is `()`, the shape of the target parameter is `(C,)`, and the shape of the output is always `()` no matter which reduction method it uses. -- 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]
