projects / platform / upstream / tvm.git / commit
? search:
summary | shortlog | log | commit | commitdiff | tree
(parent: 0a9f7e9) | patch
AutoTVM: selecting tuning templates when extracting task (#4338)
author黎明灰烬 <i@jackwish.net>
Sat, 16 Nov 2019 00:53:01 +0000 (08:53 +0800)
committerWuwei Lin <wuwei@apache.org>
Sat, 16 Nov 2019 00:53:01 +0000 (19:53 -0500)
commitccde31f1d7943376f16ab474503f8abe38a4da8b
tree5c696fbed0e01c82cce6e0f9c5c3b3413308f2e5tree | snapshot
parent0a9f7e9ade7302dd120bcf4739415a57f462148dcommit | diff
AutoTVM: selecting tuning templates when extracting task (#4338)

* AutoTVM: selecting tuning templates when extracting task

Make the procedure of trying new templates easier.

Test: tests/python/relay/test_autotvm_task_extraction.py

* Use dict to match key for topi ops

* fix lint issue

* be more pythonic :)
python/tvm/autotvm/task/relay_integration.py diff | blob | history
tests/python/relay/test_autotvm_task_extraction.py diff | blob | history
Domain: Machine Learning / ML Framework;