__init__.py 685 Bytes
Newer Older
1
#pylint: disable=wildcard-import, redefined-builtin
2 3
"""Relay core operators."""
# operator defs
4 5
from .op import get, register, register_schedule, register_compute, register_alter_op_layout, \
    Op
6
from .op import debug
7 8

# Operators
9
from .reduce import *
10
from .tensor import *
11
from .transform import *
12
from . import nn
13
from . import annotation
14 15
from . import image
from . import vision
16
from . import op_attrs
17

18

19 20
# operator registry
from . import _tensor
21
from . import _transform
22
from . import _reduce
23 24
from ..expr import Expr
from ..base import register_relay_node
25 26 27 28 29 30 31 32


def _register_op_make():
    from . import _make
    from .. import expr
    expr._op_make = _make

_register_op_make()