[Relay][Dynamic] OneHot operation (#6209)
authorMatthew Brookhart <mbrookhart@octoml.ai>
Thu, 6 Aug 2020 15:46:58 +0000 (08:46 -0700)
committerGitHub <noreply@github.com>
Thu, 6 Aug 2020 15:46:58 +0000 (08:46 -0700)
commitda75d85cdce6fa189f3662793e0a68e0f84309f1
tree9097c6518e08cc9e17ea184264ef5bb142a648a3
parent7ef89ad2dec184a033ed135dbef9fa599f55808b
[Relay][Dynamic] OneHot operation (#6209)

* Dynamic OneHot Op

* refactor dynamic_to_static

* add onehot to dynamic_to_static pass
include/tvm/topi/transform.h
python/tvm/relay/op/dyn/_transform.py
python/tvm/relay/op/transform.py
src/relay/op/dyn/tensor/transform.cc
src/relay/op/make_op.h
src/relay/transforms/dynamic_to_static.cc
tests/python/relay/dyn/test_dynamic_op_level10.py
tests/python/relay/test_pass_dynamic_to_static.py