Tianqi Chen
706f9b6f7e
* [CODEGEN/RUNTIME] Metal support, runtime improvement. * Fix case when no device is available
17 lines
325 B
ReStructuredText
17 lines
325 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
|