wehub-resource-sync
65 lines
2.0 KiB
Python
65 lines
2.0 KiB
Python
# isort: skip_file
|
|
# 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.
|
|
|
|
# pylint: disable=redefined-builtin, wildcard-import
|
|
"""TVM Operator Inventory.
|
|
|
|
TOPI is the operator collection library for TVM, to provide sugars
|
|
for constructing compute declaration as well as optimized schedules.
|
|
|
|
Some of the schedule function may have been specially optimized for a
|
|
specific workload.
|
|
"""
|
|
|
|
from tvm.libinfo import __version__
|
|
|
|
# Ensure C++ schedules get registered first, so python schedules can
|
|
# override them.
|
|
from . import cpp
|
|
|
|
from .math import *
|
|
from .tensor import *
|
|
from .index_put import *
|
|
from .reduction import *
|
|
from .transform import *
|
|
from .broadcast import *
|
|
from . import _te_tensor_overload
|
|
from .sort import *
|
|
from .scatter import *
|
|
from .scatter_elements import *
|
|
from .slice_scatter import *
|
|
from .sparse_reshape import *
|
|
from .scan import *
|
|
from .einsum import *
|
|
from .unique import *
|
|
from .searchsorted import *
|
|
from .signal import *
|
|
from . import nn
|
|
from . import utils
|
|
from . import image
|
|
from . import vision
|
|
from . import gpu
|
|
|
|
# error reporting
|
|
from .utils import InvalidShapeError
|
|
|
|
# not import testing by default
|
|
# because testing can have extra deps that are not necessary
|
|
# we can import them from test cases explicitly
|
|
# from . import testing
|