projects / platform / upstream / tvm.git / commit
? search:
summary | shortlog | log | commit | commitdiff | tree
(parent: 6b5fbda) | patch
[AutoTVM]Improve graph tuner for multiple subgraphs (#3490)
authorYao Wang <kevinthesunwy@gmail.com>
Fri, 19 Jul 2019 20:19:37 +0000 (13:19 -0700)
committerYizhi Liu <liuyizhi@apache.org>
Fri, 19 Jul 2019 20:19:37 +0000 (13:19 -0700)
commitbe26083620f5ac63db2a802c070ad9a85c2ef218
tree173ecdec55ea7583a753c1da29a4de35c004579btree | snapshot
parent6b5fbdad97708938f6fc643f7afab83693df24d8commit | diff
[AutoTVM]Improve graph tuner for multiple subgraphs (#3490)

* Improve boundary nodes in graph tuner

* Limit output node number

* Fix test

* Improve warning.

* Fix test
python/tvm/autotvm/graph_tuner/_base.py diff | blob | history
python/tvm/autotvm/graph_tuner/base_graph_tuner.py diff | blob | history
python/tvm/autotvm/graph_tuner/dynamic_programming_stage.py diff | blob | history
python/tvm/autotvm/graph_tuner/dynamic_programming_tuner.py diff | blob | history
python/tvm/autotvm/graph_tuner/pbqp_tuner.py diff | blob | history
python/tvm/autotvm/graph_tuner/utils/__init__.py diff | blob | history
python/tvm/autotvm/graph_tuner/utils/traverse_graph.py diff | blob | history
python/tvm/autotvm/graph_tuner/utils/utils.py diff | blob | history
tests/python/unittest/test_graph_tuner_core.py diff | blob | history
Domain: Machine Learning / ML Framework;