Tianqi Chen
19 lines
367 B
ReStructuredText
19 lines
367 B
ReStructuredText
tvm.ndarray
|
|
-----------
|
|
.. automodule:: tvm.ndarray
|
|
|
|
.. autoclass:: tvm.ndarray.TVMContext
|
|
:members:
|
|
|
|
.. autoclass:: tvm.ndarray.NDArray
|
|
:members:
|
|
:inherited-members:
|
|
|
|
.. autofunction:: tvm.cpu
|
|
.. autofunction:: tvm.gpu
|
|
.. autofunction:: tvm.opencl
|
|
.. autofunction:: tvm.metal
|
|
.. autofunction:: tvm.ndarray.array
|
|
|
|
.. autofunction:: tvm.register_extension
|