__init__.py 1021 Bytes
Newer Older
1
# pylint: disable=redefined-builtin, wildcard-import
2
"""TVM: Low level DSL/IR stack for tensor computation."""
tqchen committed
3
from __future__ import absolute_import as _abs
4

5
from . import tensor
6
from . import arith
7
from . import expr
tqchen committed
8 9
from . import stmt
from . import make
tqchen committed
10
from . import ir_pass
11
from . import codegen
12
from . import container
13
from . import schedule
14
from . import module
15
from . import node
16
from . import ir_builder
17
from . import target
18

19
from . import ndarray as nd
20
from .ndarray import context, cpu, gpu, opencl, cl, metal, mtl, vpi, rocm, ext_dev
21

22
from ._ffi.runtime_ctypes import TypeCode
23
from ._ffi.function import Function
24
from ._ffi.base import TVMError, __version__
25
from .api import *
26
from .intrin import *
27
from .tensor_intrin import decl_tensor_intrin
28
from .node import register_node
29
from .ndarray import register_extension
30
from .schedule import create_schedule
31
from .build_module import build, lower, build_config
32
from .tag import tag_scope
33 34 35

# Contrib initializers
from .contrib import rocm as _rocm, nvcc as _nvcc