Ubospica opened a new pull request, #14987:
URL: https://github.com/apache/tvm/pull/14987
This PR:
- Avoids inputs to nll_loss that make the total weight to be zero
- When the total weight is zero and the reduction method of nll_loss is
"mean", there would be a division by zero error
- Change `np.random.randint` into `np.random.uniform` when generating op
inputs to get better randomness
--
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]
