You are viewing a plain text version of this content. The canonical link for it is here.

Posted to commits@tvm.apache.org by GitBox <gi...@apache.org> on 2021/06/24 18:08:42 UTC

[GitHub] [tvm] vinx13 commented on pull request #8056: [Relay, TOPI] Add negative log likelihood loss (nll_loss) op

vinx13 commented on pull request #8056:
URL: https://github.com/apache/tvm/pull/8056#issuecomment-867848988


   > @vinx13 @altanh Thank you for your help!
   > 
   > > tag here is topi-level, sometimes we use it to identify a specific compute operation during schedule, otherwise we can leave it empty
   > 
   > If I change the value of `tag` to an empty string, it will fail the check in `schedule_reduce`, which is:
   > 
   > https://github.com/apache/tvm/blob/d0791d3db971a111826d96201bd1e4c9c0d531da/python/tvm/topi/x86/reduction.py#L84
   > 
   > I'm not sure if I need to adjust somewhere else...
   
   in this case we need to change tag to `kInjective` as the reduction op is not broadcast


-- 
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:
users@infra.apache.org