Tianqi Chen
e0122c0ea6
* [REFACTOR][PY-API] Polish tvm.runtime, tvm.runtime.module API update This PR updates the tvm.runtime to use the new FFI style. - Remove top-level tvm.module to avoid confusion between runtime.Module and IRModule - API changes wrt to runtime.Module - tvm.module.load -> tvm.runtime.load_module - tvm.module.enabled -> tvm.runtime.enabled - tvm.module.system_lib -> tvm.runtime.system_lib - Remove dep on api_internal from runtime. * Update module.load in the latest API
37 lines
1.2 KiB
ReStructuredText
37 lines
1.2 KiB
ReStructuredText
.. Licensed to the Apache Software Foundation (ASF) under one
|
|
or more contributor license agreements. See the NOTICE file
|
|
distributed with this work for additional information
|
|
regarding copyright ownership. The ASF licenses this file
|
|
to you under the Apache License, Version 2.0 (the
|
|
"License"); you may not use this file except in compliance
|
|
with the License. You may obtain a copy of the License at
|
|
|
|
.. http://www.apache.org/licenses/LICENSE-2.0
|
|
|
|
.. Unless required by applicable law or agreed to in writing,
|
|
software distributed under the License is distributed on an
|
|
"AS IS" BASIS, WITHOUT WARRANTIES OR CONDITIONS OF ANY
|
|
KIND, either express or implied. See the License for the
|
|
specific language governing permissions and limitations
|
|
under the License.
|
|
|
|
tvm.runtime.ndarray
|
|
-------------------
|
|
.. automodule:: tvm.runtime.ndarray
|
|
|
|
.. autoclass:: tvm.nd.NDArray
|
|
:members:
|
|
:inherited-members:
|
|
|
|
|
|
.. autoclass:: tvm.runtime.TVMContext
|
|
:members:
|
|
|
|
.. autofunction:: tvm.context
|
|
.. autofunction:: tvm.cpu
|
|
.. autofunction:: tvm.gpu
|
|
.. autofunction:: tvm.opencl
|
|
.. autofunction:: tvm.metal
|
|
.. autofunction:: tvm.nd.array
|
|
.. autofunction:: tvm.nd.empty
|