[Relay] crossentropy_with_logits and its gradient (#4075)
author雾雨魔理沙 <lolisa@marisa.moe>
Fri, 25 Oct 2019 14:54:29 +0000 (07:54 -0700)
committerThierry Moreau <moreau@uw.edu>
Fri, 25 Oct 2019 14:54:29 +0000 (17:54 +0300)
commit1ad6a2af672dbb94861d00985f4b9f8280d04d6e
tree7f094489aa1466234a1369e02bcd98ac12fe07b5
parent493c98d3020fefa9f6dd3c291a9eca674838d520
[Relay] crossentropy_with_logits and its gradient (#4075)

* save

* lint
python/tvm/relay/op/_reduce.py
python/tvm/relay/op/_tensor_grad.py
python/tvm/relay/op/nn/_nn.py
python/tvm/relay/op/nn/nn.py
src/relay/op/nn/nn.cc
tests/python/relay/test_op_grad_level10.py