pytorch
8bbe3838 - [Static Runtime] Fix bugs in logit (#57578)

Commit
3 years ago
[Static Runtime] Fix bugs in logit (#57578) Summary: Pull Request resolved: https://github.com/pytorch/pytorch/pull/57578 The original impl in SR assumes that eps is a constant, which is true most of the times. However it could be a graph input as well. This diff fixes this issue. Unit tests are added as well. Reviewed By: edvgha Differential Revision: D28207975 fbshipit-source-id: 9a10dec159f3804e43ef74aaa20c3ec6c79548c9
Author
Hao Lu
Parents
Loading