Add operation scatter_add to relay, based on scatter implementation. (#6030)
authornotoraptor <notoraptor@users.noreply.github.com>
Thu, 16 Jul 2020 02:21:28 +0000 (22:21 -0400)
committerGitHub <noreply@github.com>
Thu, 16 Jul 2020 02:21:28 +0000 (19:21 -0700)
commit3912a64e43412704fcd0c170e94a49dcc6e62f71
treeee4f148cbaee8ad480211569d2d91979d6ea8ef0
parent6360ad1583193b4d1b7bd910560a5e95258a39ab
Add operation scatter_add to relay, based on scatter implementation. (#6030)
include/tvm/relay/attrs/transform.h
python/tvm/relay/op/_transform.py
python/tvm/relay/op/strategy/generic.py
python/tvm/relay/op/transform.py
src/relay/op/tensor/transform.cc
tests/python/relay/test_op_level3.py
topi/python/topi/__init__.py
topi/python/topi/generic/search.py
topi/python/topi/scatter_add.py [new file with mode: 0644]