[relay][pass] Annotation for heterogeneous compilation (#2361)
Showing
- docs/langref/relay_op.rst 2 additions, 0 deletionsdocs/langref/relay_op.rst
- include/tvm/relay/attrs/annotation.h 31 additions, 0 deletionsinclude/tvm/relay/attrs/annotation.h
- include/tvm/relay/attrs/device_copy.h 36 additions, 0 deletionsinclude/tvm/relay/attrs/device_copy.h
- include/tvm/relay/attrs/transform.h 0 additions, 1 deletioninclude/tvm/relay/attrs/transform.h
- include/tvm/relay/pass.h 15 additions, 0 deletionsinclude/tvm/relay/pass.h
- python/tvm/__init__.py 1 addition, 0 deletionspython/tvm/__init__.py
- python/tvm/relay/__init__.py 1 addition, 0 deletionspython/tvm/relay/__init__.py
- python/tvm/relay/annotation.py 4 additions, 0 deletionspython/tvm/relay/annotation.py
- python/tvm/relay/backend/_backend.py 6 additions, 5 deletionspython/tvm/relay/backend/_backend.py
- python/tvm/relay/backend/graph_runtime_codegen.py 68 additions, 15 deletionspython/tvm/relay/backend/graph_runtime_codegen.py
- python/tvm/relay/build_module.py 155 additions, 19 deletionspython/tvm/relay/build_module.py
- python/tvm/relay/ir_pass.py 57 additions, 0 deletionspython/tvm/relay/ir_pass.py
- python/tvm/relay/op/__init__.py 1 addition, 0 deletionspython/tvm/relay/op/__init__.py
- python/tvm/relay/op/annotation/__init__.py 4 additions, 0 deletionspython/tvm/relay/op/annotation/__init__.py
- python/tvm/relay/op/annotation/_make.py 4 additions, 0 deletionspython/tvm/relay/op/annotation/_make.py
- python/tvm/relay/op/annotation/annotation.py 31 additions, 0 deletionspython/tvm/relay/op/annotation/annotation.py
- python/tvm/relay/op/tensor.py 41 additions, 0 deletionspython/tvm/relay/op/tensor.py
- src/relay/backend/compile_engine.cc 39 additions, 10 deletionssrc/relay/backend/compile_engine.cc
- src/relay/backend/graph_plan_memory.cc 40 additions, 17 deletionssrc/relay/backend/graph_plan_memory.cc
- src/relay/op/annotation/annotation.cc 42 additions, 0 deletionssrc/relay/op/annotation/annotation.cc
include/tvm/relay/attrs/annotation.h
0 → 100644
include/tvm/relay/attrs/device_copy.h
0 → 100644
python/tvm/relay/annotation.py
0 → 100644
python/tvm/relay/op/annotation/__init__.py
0 → 100644
python/tvm/relay/op/annotation/_make.py
0 → 100644
python/tvm/relay/op/annotation/annotation.py
0 → 100644
src/relay/op/annotation/annotation.cc
0 → 100644
Please register or sign in to comment