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

projects / platform / upstream / tvm.git / commit

author	Haichen Shen <shenhaichen@gmail.com>
	Tue, 4 Jun 2019 23:29:56 +0000 (16:29 -0700)
committer	Leyuan Wang <laurawly@gmail.com>
	Tue, 4 Jun 2019 23:29:56 +0000 (16:29 -0700)
commit	072f8cc75ec435f6c9557b480232018c9e22c25f
tree	16df96821b2f00ade9f8953eecda4e8e85e2eb23	tree \| snapshot
parent	4204544b8b4b52305cd57d81e7755348f2acba51	commit \| diff

[Relay/TOPI][Op] Add TopK operator (#3256)

* init impl for topk

* Fix cpu for topk

* init cuda impl for topk

* Add cuda for topk

* fix

* Add doc

* update doc

* lint

* lint

* lint

* x

* fix warning

* [Relay] Add TopK in tf converter

* Add frontend converter

* fix

24 files changed:

docs/api/python/topi.rst		diff \| blob \| history
docs/langref/relay_op.rst		diff \| blob \| history
include/tvm/relay/attrs/algorithm.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/_algorithm.py		diff \| blob \| history
python/tvm/relay/op/algorithm.py		diff \| blob \| history
python/tvm/relay/op/nn/nn.py		diff \| blob \| history
python/tvm/relay/op/transform.py		diff \| blob \| history
src/codegen/build_module.cc		diff \| blob \| history
src/contrib/sort/sort.cc		diff \| blob \| history
src/relay/op/algorithm/argsort.cc	[moved from src/relay/op/algorithm/sort.cc with 94% similarity]	diff \| blob \| history
src/relay/op/algorithm/topk.cc	[new file with mode: 0644]	blob
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_level6.py		diff \| blob \| history
topi/python/topi/cuda/__init__.py		diff \| blob \| history
topi/python/topi/cuda/nms.py		diff \| blob \| history
topi/python/topi/cuda/sort.py		diff \| blob \| history
topi/python/topi/generic/sort.py		diff \| blob \| history
topi/python/topi/sort.py		diff \| blob \| history
topi/python/topi/transform.py		diff \| blob \| history
topi/python/topi/vision/nms.py		diff \| blob \| history
topi/tests/python/test_topi_sort.py		diff \| blob \| history

Domain: Machine Learning / ML Framework;

RSS Atom