tvm.rst 852 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 18

.. 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
   tvm.thread_axis
19
   tvm.comm_reducer
20 21 22
   tvm.sum
   tvm.min
   tvm.max
23
   tvm.tag_scope
24 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
.. autofunction:: tvm.thread_axis
37
.. autofunction:: tvm.comm_reducer
38 39 40
.. autofunction:: tvm.sum
.. autofunction:: tvm.min
.. autofunction:: tvm.max
41
.. autofunction:: tvm.tag_scope