mindspore 2.1.0__cp39-cp39-win_amd64.whl → 2.2.10__cp39-cp39-win_amd64.whl

mindspore/_extends/graph_kernel/expanders/softsign.py

@@ -1,28 +0,0 @@
-# Copyright 2020-2021 Huawei Technologies Co., Ltd
-#
-# Licensed 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.
-# ===========================================================================
-"""generate json desc for Softsign"""
-from ._utils import Expander
-
-
-class Softsign(Expander):
-    """Softsign expander"""
-
-    def _expand(self, graph_builder):
-        x = self.inputs[0]
-        x_abs = graph_builder.emit('Abs', [x])
-        one = graph_builder.value(x.dtype, 1.0)
-        sum_x = graph_builder.emit('Add', [x_abs, one])
-        result = graph_builder.emit('RealDiv', [x, sum_x])
-        return result

mindspore/_extends/graph_kernel/expanders/sqrt_grad.py

@@ -1,29 +0,0 @@
-# Copyright 2020-2021 Huawei Technologies Co., Ltd
-#
-# Licensed 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.
-# ===========================================================================
-"""generate json desc for sqrtgrad"""
-from ._utils import Expander, ExpanderInfoValidator as VLD
-
-
-@VLD.check_all_formats_same
-class SqrtGrad(Expander):
-    """SqrtGrad expander"""
-
-    def _expand(self, graph_builder):
-        # formula of sqrt_grad is dout / (2 * x)
-        x, dout = self.inputs
-        const_two = graph_builder.value(x.dtype, 2)
-        dividend = graph_builder.emit('Mul', [x, const_two])
-        result = graph_builder.emit('RealDiv', [dout, dividend])
-        return result

mindspore/_extends/graph_kernel/expanders/square_sum_all.py

@@ -1,44 +0,0 @@
-# Copyright 2021-2022 Huawei Technologies Co., Ltd
-#
-# Licensed 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.
-# ===========================================================================
-"""generate json desc for SquareSumAll"""
-from mindspore._extends.graph_kernel.model.model import GraphKernelUnsupportedException as GKException
-from ._utils import Expander
-
-
-class SquareSumAll(Expander):
-    """SquareSumAll expander"""
-
-    def _check(self):
-        """check inputs"""
-        input_num = len(self.inputs)
-        if input_num != 2:
-            raise GKException("For 'SquareSumAll', the inputs number should be 2, but got {}.".format(input_num))
-
-    def _expand(self, graph_builder):
-        """do expand"""
-        x0 = self.inputs[0]
-        x1 = self.inputs[1]
-
-        ori_shape = x0.shape
-        axis = []
-        for i, _ in enumerate(ori_shape):
-            axis.append(i)
-
-        square_res0 = graph_builder.emit('Mul', [x0, x0])
-        square_res1 = graph_builder.emit('Mul', [x1, x1])
-        result0 = graph_builder.emit('ReduceSum', [square_res0], attrs={'reduce_axis': axis, 'keep_dims': False})
-        result1 = graph_builder.emit('ReduceSum', [square_res1], attrs={'reduce_axis': axis, 'keep_dims': False})
-
-        return result0, result1

mindspore/_extends/graph_kernel/expanders/square_sum_v1.py

@@ -1,37 +0,0 @@
-# Copyright 2021-2022 Huawei Technologies Co., Ltd
-#
-# Licensed 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.
-# ===========================================================================
-"""generate json desc for SquareSumV1"""
-from mindspore._extends.graph_kernel.model.model import DataFormat as DF
-from ._utils import Expander, ExpanderInfoValidator as VLD
-from ._utils import get_reduce_axis_shape
-
-
-@VLD.add_format(DF.FRAC_NZ)
-@VLD.add_format(DF.DEFAULT)
-@VLD.check_attrs('axis')
-class SquareSumV1(Expander):
-    """Square expander"""
-
-    def _expand(self, graph_builder):
-        x = self.inputs[0]
-        axis = self.attrs['axis']
-
-        reduce_axis, ori_reduced_shape = get_reduce_axis_shape(x.shape, x.data_format, axis)
-
-        square_res = graph_builder.emit('Mul', [x, x])
-        result = graph_builder.emit('ReduceSum', [square_res], attrs={'reduce_axis': reduce_axis, 'keep_dims': False})
-        if x.data_format == DF.FRAC_NZ:
-            result = graph_builder.emit('Reshape', [result], attrs={'shape': ori_reduced_shape})
-        return result

mindspore/_extends/graph_kernel/expanders/squared_difference.py

@@ -1,43 +0,0 @@
-# Copyright 2021 Huawei Technologies Co., Ltd
-#
-# Licensed 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.
-# ===========================================================================
-"""generate json desc for squared_difference"""
-from mindspore._extends.graph_kernel.model.model import GraphKernelUnsupportedException as GKException
-from ._utils import Expander, ExpanderInfoValidator as VLD
-
-
-@VLD.check_all_formats_same
-class SquaredDifference(Expander):
-    """SquaredDifference expander"""
-
-    def __init__(self, expand_info):
-        super().__init__(expand_info)
-        self.dtype_x = self.inputs[0]['data_type']
-        self.dtype_y = self.inputs[1]['data_type']
-
-    def _check(self):
-        if self.dtype_x == "float64" or self.dtype_y == "float64":
-            raise GKException("For 'SquaredDifference', the inputs data type must not be float64")
-        if self.dtype_x != self.dtype_y:
-            raise GKException("For 'SquaredDifference', the inputs data type should be same, but got {} and {}"
-                              .format(self.dtype_x, self.dtype_y))
-
-    def _expand(self, graph_builder):
-        input_x = self.inputs[0]
-        input_y = self.inputs[1]
-
-        sub_val = graph_builder.emit('Sub', [input_x, input_y])
-        result = graph_builder.emit('Mul', [sub_val, sub_val])
-
-        return result

mindspore/_extends/graph_kernel/expanders/tanh_grad.py

@@ -1,31 +0,0 @@
-# Copyright 2020-2021 Huawei Technologies Co., Ltd
-#
-# Licensed 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.
-# ===========================================================================
-"""generate json desc for tanh_grad"""
-from ._utils import Expander, ExpanderInfoValidator as VLD
-
-
-@VLD.check_all_formats_same
-class TanhGrad(Expander):
-    """TanhGrad expander"""
-
-    def _expand(self, graph_builder):
-        input_y, input_dy = self.inputs
-
-        const_one = graph_builder.value(input_y.dtype, 1)
-        double_y = graph_builder.emit('Mul', [input_y, input_y])
-        one_sub_double_y = graph_builder.emit('Sub', [const_one, double_y])
-        result = graph_builder.emit('Mul', [input_dy, one_sub_double_y])
-
-        return result