# 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