[Relay][Training] Add gradient for Crossentropy (#3925)
author雾雨魔理沙 <lolisa@marisa.moe>
Sat, 5 Oct 2019 00:24:55 +0000 (17:24 -0700)
committerWuwei Lin <wuwei@apache.org>
Sat, 5 Oct 2019 00:24:55 +0000 (20:24 -0400)
commit7d71dd8bd7b9b6412568a6860161566071588055
treef2b585f2481a983617dc54d6647089052ce77d58
parent59d8d4006bf984e1b7f9255df1f10334f12738ef
[Relay][Training] Add gradient for Crossentropy (#3925)

* save

save

redo max test

save

address comment

fix

* address comment

* increase rtol

* address review comment
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
python/tvm/relay/testing/__init__.py
src/relay/op/nn/nn.cc
tests/python/relay/test_op_grad_level10.py [new file with mode: 0644]
tests/python/relay/test_op_grad_level4.py
tests/python/relay/test_op_level5.py