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""" 17Composite operators. 18 19Pre-defined combination of operators. 20""" 21 22 23from .base import GradOperation, HyperMap, Map, MultitypeFuncGraph, add_flags, \ 24 core, env_get, tail, zip_operation 25from .clip_ops import clip_by_value, clip_by_global_norm 26from .multitype_ops.add_impl import hyper_add 27from .multitype_ops.ones_like_impl import ones_like 28from .multitype_ops.zeros_like_impl import zeros_like 29from .random_ops import normal, laplace, uniform, gamma, poisson, multinomial 30from .math_ops import count_nonzero, tensor_dot, dot, batch_dot, matmul 31from .array_ops import repeat_elements, sequence_mask 32 33 34__all__ = [ 35 'env_get', 36 'core', 37 'add_flags', 38 'tail', 39 'MultitypeFuncGraph', 40 'GradOperation', 41 'HyperMap', 42 'hyper_add', 43 'zeros_like', 44 'ones_like', 45 'zip_operation', 46 'normal', 47 'laplace', 48 'uniform', 49 'gamma', 50 'poisson', 51 'multinomial', 52 'clip_by_value', 53 'clip_by_global_norm', 54 'count_nonzero', 55 'tensor_dot', 56 'dot', 57 'batch_dot', 58 'repeat_elements', 59 'sequence_mask', 60 'matmul'] 61