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/05/19 02:39:11 UTC

[GitHub] [tvm] zhuzilin edited a comment on pull request #8056: [Relay, TOPI] Add negative log likelihood loss (nll_loss) op

zhuzilin edited a comment on pull request #8056:
URL: https://github.com/apache/tvm/pull/8056#issuecomment-843696836


   @tkonolige Sorry about the incomplete formula... I don't know what  I was thinking about yesterday :scream:
   
   > are you planning on registering a gradient for this operator?
   
   @altanh Yes, I'm trying to  write one for this op. I will try to implement one just like the cpu version of pytorch (https://github.com/pytorch/pytorch/blob/2ddd8416350b701fe0e7325e4761f8d14a3f46e5/aten/src/ATen/native/LossNLL.cpp#L29). It may take some time since it's a little complex...


-- 
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