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 like the cpu version of pytorch (https://github.com/pytorch/pytorch/blob/2ddd8416350b701fe0e7325e4761f8d14a3f46e5/aten/src/ATen/native/LossNLL.cpp#L29 and https://github.com/pytorch/pytorch/blob/2ddd8416350b701fe0e7325e4761f8d14a3f46e5/aten/src/ATen/native/LossNLL2d.cpp#L253). 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: [email protected]
