projects / platform / upstream / tvm.git / commit
? search:
summary | shortlog | log | commit | commitdiff | tree
(parent: 6e085b4) | patch
Fix bias_add gradient (#4516)
authorSWu <SWu@users.noreply.github.com>
Fri, 13 Dec 2019 20:09:56 +0000 (15:09 -0500)
committerWuwei Lin <wuwei@apache.org>
Fri, 13 Dec 2019 20:09:56 +0000 (15:09 -0500)
commitf10944c90c93d99e1854969d6d19e9b30fae6a8b
tree068de8fbad74b3b272eb99d8f03113e9368c38bdtree | snapshot
parent6e085b40328a99ec59a7e4ff50017edab31eb553commit | diff
Fix bias_add gradient (#4516)

* Fix bias_add gradient

A change caused collapse_sum_like to reject implicit dimension
broadcasting for bias_add gradient, so switch to explicit sum reduction
on the non-bias axis dimensions.

* Lint fix
python/tvm/relay/op/_tensor_grad.py diff | blob | history
tests/python/relay/test_op_grad_level1.py diff | blob | history
Domain: Machine Learning / ML Framework;