1# Copyright 2020 Huawei Technologies Co., Ltd 2# 3# Licensed under the Apache License, Version 2.0 (the "License"); 4# you may not use this file except in compliance with the License. 5# You may obtain a copy of the License at 6# 7# http://www.apache.org/licenses/LICENSE-2.0 8# 9# Unless required by applicable law or agreed to in writing, software 10# distributed under the License is distributed on an "AS IS" BASIS, 11# WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied. 12# See the License for the specific language governing permissions and 13# limitations under the License. 14# ============================================================================ 15 16""" 17Operators can be used in the construct function of Cell. 18 19Examples: 20 21 >>> import mindspore.ops as ops 22""" 23 24from .primitive import Primitive, PrimitiveWithInfer, PrimitiveWithCheck, prim_attr_register 25from .vm_impl_registry import get_vm_impl_fn, vm_impl_registry 26from .op_info_register import op_info_register, AkgGpuRegOp, AkgAscendRegOp, AiCPURegOp, TBERegOp, CpuRegOp, DataType 27from .primitive import constexpr 28from . import composite, operations, functional 29from . import signature 30from .composite import * 31from .operations import * 32from .functional import * 33 34__primitive__ = [ 35 "prim_attr_register", "Primitive", "PrimitiveWithInfer", "PrimitiveWithCheck", "signature" 36] 37 38__all__ = ["get_vm_impl_fn", "vm_impl_registry", 39 "op_info_register", "AkgGpuRegOp", "AkgAscendRegOp", "AiCPURegOp", "TBERegOp", "CpuRegOp", "DataType", 40 "constexpr"] 41__all__.extend(__primitive__) 42__all__.extend(composite.__all__) 43__all__.extend(operations.__all__) 44__all__.extend(functional.__all__) 45