-
- Downloads
[TAG] Add tvm.tag module for tagging operator (#217)
* [TAG] Add op_tag module for tagging operator * Fix accroading to comments * Add example * Add into doc * Add --fix-missing for docker
Showing
- docs/api/python/tvm.rst 2 additions, 0 deletionsdocs/api/python/tvm.rst
- include/tvm/operation.h 33 additions, 11 deletionsinclude/tvm/operation.h
- python/tvm/__init__.py 1 addition, 0 deletionspython/tvm/__init__.py
- python/tvm/api.py 21 additions, 8 deletionspython/tvm/api.py
- python/tvm/tag.py 65 additions, 0 deletionspython/tvm/tag.py
- src/api/api_lang.cc 6 additions, 3 deletionssrc/api/api_lang.cc
- src/op/compute_op.cc 13 additions, 5 deletionssrc/op/compute_op.cc
- src/op/extern_op.cc 2 additions, 0 deletionssrc/op/extern_op.cc
- src/op/scan_op.cc 5 additions, 2 deletionssrc/op/scan_op.cc
- src/schedule/schedule_dataflow_rewrite.cc 3 additions, 3 deletionssrc/schedule/schedule_dataflow_rewrite.cc
- tests/ci_build/Dockerfile.cpu 1 addition, 1 deletiontests/ci_build/Dockerfile.cpu
- tests/ci_build/Dockerfile.i386 1 addition, 1 deletiontests/ci_build/Dockerfile.i386
- tests/python/unittest/test_lang_tag.py 63 additions, 0 deletionstests/python/unittest/test_lang_tag.py
Loading
Please register or sign in to comment