review.tizen.org Git - platform/upstream/tvm.git/commit

projects / platform / upstream / tvm.git / commit

author	Jon Soifer <soiferj@gmail.com>
	Thu, 22 Aug 2019 20:45:45 +0000 (13:45 -0700)
committer	Haichen Shen <shenhaichen@gmail.com>
	Thu, 22 Aug 2019 20:45:45 +0000 (13:45 -0700)
commit	554df2118931cede5c0783c5a41d4da0be46ffe6
tree	e2d4b3ceff7873dea22be0943df1d2a7837cd2dc	tree \| snapshot
parent	7264cb6a5996af874819f03cf27aca3b5ad48814	commit \| diff

[TOPI][Relay][TensorFlow] Add OneHot operator (#3781)

* Add one-hot to Relay

* topi implementation

* Working

* add topi test

* Add TF test

* Fix check

* fix linting issues

* fix documentation

* Fix documentation

* Add support for on_value, off_value, axis, dtype

* Add full support for axis

* Fix compute and update test_forward

* Move on_value and off_value to inputs

* Add topi test

* Update tests

* Update docs

* Fix style

* re-enable tests

* Add one_hot to mxnet converter

17 files changed:

docs/api/python/topi.rst		diff \| blob \| history
docs/langref/relay_op.rst		diff \| blob \| history
include/tvm/relay/attrs/transform.h		diff \| blob \| history
python/tvm/relay/frontend/mxnet.py		diff \| blob \| history
python/tvm/relay/frontend/tensorflow.py		diff \| blob \| history
python/tvm/relay/op/_transform.py		diff \| blob \| history
python/tvm/relay/op/transform.py		diff \| blob \| history
src/relay/op/tensor/transform.cc		diff \| blob \| history
tests/python/frontend/mxnet/test_forward.py		diff \| blob \| history
tests/python/frontend/tensorflow/test_forward.py		diff \| blob \| history
tests/python/relay/test_op_level10.py		diff \| blob \| history
topi/include/topi/transform.h		diff \| blob \| history
topi/python/topi/testing/__init__.py		diff \| blob \| history
topi/python/topi/testing/one_hot.py	[new file with mode: 0644]	blob
topi/python/topi/transform.py		diff \| blob \| history
topi/src/topi.cc		diff \| blob \| history
topi/tests/python/test_topi_transform.py		diff \| blob \| history

Domain: Machine Learning / ML Framework;

RSS Atom