tvm.rst 895 Bytes
Newer Older
1 2
tvm
---
3
The user facing API for computation declaration.
4 5 6 7 8 9 10 11 12 13 14 15 16 17

.. autosummary::

   tvm.load_json
   tvm.save_json
   tvm.var
   tvm.const
   tvm.convert
   tvm.placeholder
   tvm.compute
   tvm.scan
   tvm.extern
   tvm.decl_buffer
   tvm.reduce_axis
18
   tvm.select
19
   tvm.thread_axis
20
   tvm.comm_reducer
21 22 23
   tvm.sum
   tvm.min
   tvm.max
24
   tvm.tag_scope
25 26 27 28 29 30 31 32 33 34 35 36

.. autofunction:: tvm.load_json
.. autofunction:: tvm.save_json
.. autofunction:: tvm.var
.. autofunction:: tvm.const
.. autofunction:: tvm.convert
.. autofunction:: tvm.placeholder
.. autofunction:: tvm.compute
.. autofunction:: tvm.scan
.. autofunction:: tvm.extern
.. autofunction:: tvm.decl_buffer
.. autofunction:: tvm.reduce_axis
37
.. autofunction:: tvm.select
38
.. autofunction:: tvm.thread_axis
39
.. autofunction:: tvm.comm_reducer
40 41 42
.. autofunction:: tvm.sum
.. autofunction:: tvm.min
.. autofunction:: tvm.max
43
.. autofunction:: tvm.tag_scope