Add Scatter to Topi/Relay/ONNX via hybrid script (#5619)
authorMatthew Brookhart <mbrookhart@octoml.ai>
Tue, 9 Jun 2020 22:33:57 +0000 (15:33 -0700)
committerGitHub <noreply@github.com>
Tue, 9 Jun 2020 22:33:57 +0000 (07:33 +0900)
commit9f79199822232ccb9d0b18cd9250dccc22fa3dd0
tree67e0daa185bcdf09aa6ef7e1e90bce0fe1cc4d26
parentaa808570dbee19a178130c0f7d9b397ff7f51f0e
Add Scatter to Topi/Relay/ONNX via hybrid script (#5619)

* I can construct scatter but not embed it in a Relay Graph

* working 1-4 dimesion scatter

* add scatter to ONNX

fix lint

* isolate tests to cpu backend

* Fix i386 test

* fix gpu tolerance

* use elemwise_shape_func for scatter

* fix incorrect rebase
12 files changed:
include/tvm/relay/attrs/transform.h
python/tvm/relay/frontend/onnx.py
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/frontend/onnx/test_forward.py
tests/python/relay/test_op_level3.py
tests/python/relay/test_op_level5.py
topi/python/topi/__init__.py
topi/python/topi/generic/search.py
topi/python/topi/scatter.py [new file with mode: 0644]