tree: 0357c83b6a8f45cf1f66a62bb903727dbe05ec0f [path history] [tgz]
  1. api/
  2. arithmetic/
  3. autotvm/
  4. codegen/
  5. common/
  6. contrib/
  7. lang/
  8. op/
  9. pass/
  10. relay/
  11. runtime/
  12. schedule/
  13. README.md
src/README.md

Code Organization

Header files in include are public APIs that share across modules. There can be internal header files within each module that sit in src.

Modules

  • common: Internal common utilities.
  • api: API function registration.
  • lang: The definition of DSL related data structure.
  • arithmetic: Arithmetic expression and set simplification.
  • op: The detail implementations about each operation(compute, scan, placeholder).
  • schedule: The operations on the schedule graph before converting to IR.
  • pass: The optimization pass on the IR structure.
  • codegen: The code generator.
  • runtime: Minimum runtime related codes.
  • autotvm: The auto-tuning module.
  • relay: Implementation of Relay. The second generation of NNVM, a new IR for deep learning frameworks.
  • contrib: Contrib extension libraries.