ndarray.rst 367 Bytes
Newer Older
1 2
tvm.ndarray
-----------
3
.. automodule:: tvm.ndarray
4 5 6 7 8 9 10 11 12 13 14

.. autoclass:: tvm.ndarray.TVMContext
    :members:

.. autoclass:: tvm.ndarray.NDArray
    :members:
    :inherited-members:

.. autofunction:: tvm.cpu
.. autofunction:: tvm.gpu
.. autofunction:: tvm.opencl
15
.. autofunction:: tvm.metal
16
.. autofunction:: tvm.ndarray.array
17

18
.. autofunction:: tvm.register_extension