mindspore 2.2.14__cp39-cp39-win_amd64.whl → 2.3.0__cp39-cp39-win_amd64.whl

mindspore/ops/_op_impl/tbe/sigmoid_ds.py

@@ -1,38 +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.
-# ============================================================================
-
-"""Sigmoid op"""
-from mindspore.ops.op_info_register import op_info_register, TBERegOp, DataType
-
-sigmoid_op_info = TBERegOp("Sigmoid") \
-    .fusion_type("ELEMWISE") \
-    .async_flag(False) \
-    .binfile_name("sigmoid.so") \
-    .compute_cost(10) \
-    .kernel_name("sigmoid") \
-    .partial_flag(True) \
-    .dynamic_shape(True) \
-    .input(0, "x", False, "required", "all") \
-    .output(0, "y", False, "required", "all") \
-    .op_pattern("formatAgnostic") \
-    .dtype_format(DataType.F16_None, DataType.F16_None) \
-    .dtype_format(DataType.F32_None, DataType.F32_None) \
-    .get_op_info()
-
-
-@op_info_register(sigmoid_op_info)
-def _sigmoid_ds_tbe():
-    """Sigmoid TBE register"""
-    return

mindspore/ops/_op_impl/tbe/sigmoid_grad.py

@@ -1,39 +0,0 @@
-# Copyright 2020 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.
-# ============================================================================
-
-"""SigmoidGrad op"""
-from mindspore.ops.op_info_register import op_info_register, TBERegOp, DataType
-
-sigmoid_grad_op_info = TBERegOp("SigmoidGrad") \
-    .fusion_type("OPAQUE") \
-    .async_flag(False) \
-    .binfile_name("sigmoid_grad.so") \
-    .compute_cost(10) \
-    .kernel_name("sigmoid_grad") \
-    .partial_flag(True) \
-    .input(0, "x", False, "required", "all") \
-    .input(1, "y", False, "required", "all") \
-    .output(0, "z", False, "required", "all") \
-    .dtype_format(DataType.F16_Default, DataType.F16_Default, DataType.F16_Default) \
-    .dtype_format(DataType.F16_5HD, DataType.F16_5HD, DataType.F16_5HD) \
-    .dtype_format(DataType.F32_Default, DataType.F32_Default, DataType.F32_Default) \
-    .dtype_format(DataType.F32_5HD, DataType.F32_5HD, DataType.F32_5HD) \
-    .get_op_info()
-
-
-@op_info_register(sigmoid_grad_op_info)
-def _sigmoid_grad_tbe():
-    """SigmoidGrad TBE register"""
-    return

mindspore/ops/_op_impl/tbe/sigmoid_grad_ds.py

@@ -1,40 +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.
-# ============================================================================
-
-"""SigmoidGrad op"""
-from mindspore.ops.op_info_register import op_info_register, TBERegOp, DataType
-
-sigmoid_grad_ds_op_info = TBERegOp("SigmoidGrad") \
-    .fusion_type("ELEMWISE") \
-    .async_flag(False) \
-    .binfile_name("sigmoid_grad.so") \
-    .compute_cost(10) \
-    .kernel_name("sigmoid_grad") \
-    .partial_flag(True) \
-    .dynamic_shape(True) \
-    .input(0, "y", False, "required", "all") \
-    .input(1, "dy", False, "required", "all") \
-    .output(0, "output", False, "required", "all") \
-    .dtype_format(DataType.F16_Default, DataType.F16_Default, DataType.F16_Default) \
-    .dtype_format(DataType.F16_5HD, DataType.F16_5HD, DataType.F16_5HD) \
-    .dtype_format(DataType.F32_Default, DataType.F32_Default, DataType.F32_Default) \
-    .dtype_format(DataType.F32_5HD, DataType.F32_5HD, DataType.F32_5HD) \
-    .get_op_info()
-
-
-@op_info_register(sigmoid_grad_ds_op_info)
-def _sigmoid_grad_ds_tbe():
-    """SigmoidGrad TBE register"""
-    return

mindspore/ops/_op_impl/tbe/sign.py

@@ -1,38 +0,0 @@
-# Copyright 2020 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.
-# ============================================================================
-
-"""Sign op"""
-from mindspore.ops.op_info_register import op_info_register, TBERegOp, DataType
-
-sign_op_info = TBERegOp("Sign") \
-    .fusion_type("ELEMWISE") \
-    .async_flag(False) \
-    .binfile_name("sign.so") \
-    .compute_cost(10) \
-    .kernel_name("sign") \
-    .partial_flag(True) \
-    .op_pattern("formatAgnostic") \
-    .input(0, "x", None, "required", None) \
-    .output(0, "y", True, "required", "all") \
-    .dtype_format(DataType.F16_None, DataType.F16_None) \
-    .dtype_format(DataType.F32_None, DataType.F32_None) \
-    .dtype_format(DataType.I32_None, DataType.I32_None) \
-    .get_op_info()
-
-
-@op_info_register(sign_op_info)
-def _sign_tbe():
-    """Sign TBE register"""
-    return

mindspore/ops/_op_impl/tbe/sign_ds.py

@@ -1,39 +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.
-# ============================================================================
-
-"""Sign op"""
-from mindspore.ops.op_info_register import op_info_register, TBERegOp, DataType
-
-sign_ds_op_info = TBERegOp("Sign") \
-    .fusion_type("ELEMWISE") \
-    .async_flag(False) \
-    .binfile_name("sign.so") \
-    .compute_cost(10) \
-    .kernel_name("sign") \
-    .partial_flag(True) \
-    .dynamic_shape(True) \
-    .op_pattern("formatAgnostic") \
-    .input(0, "x", None, "required", None) \
-    .output(0, "y", True, "required", "all") \
-    .dtype_format(DataType.F16_None, DataType.F16_None) \
-    .dtype_format(DataType.F32_None, DataType.F32_None) \
-    .dtype_format(DataType.I32_None, DataType.I32_None) \
-    .get_op_info()
-
-
-@op_info_register(sign_ds_op_info)
-def _sign_ds_tbe():
-    """Sign TBE register"""
-    return

mindspore/ops/_op_impl/tbe/sin.py

@@ -1,37 +0,0 @@
-# Copyright 2020 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.
-# ============================================================================
-
-"""Sin op"""
-from mindspore.ops.op_info_register import op_info_register, TBERegOp, DataType
-
-sin_op_info = TBERegOp("Sin") \
-    .fusion_type("ELEMWISE") \
-    .async_flag(False) \
-    .binfile_name("sin.so") \
-    .compute_cost(10) \
-    .kernel_name("sin") \
-    .partial_flag(True) \
-    .op_pattern("formatAgnostic") \
-    .input(0, "x", False, "required", "all") \
-    .output(0, "y", False, "required", "all") \
-    .dtype_format(DataType.F16_None, DataType.F16_None) \
-    .dtype_format(DataType.F32_None, DataType.F32_None) \
-    .get_op_info()
-
-
-@op_info_register(sin_op_info)
-def _sin_tbe():
-    """Sin TBE register"""
-    return

mindspore/ops/_op_impl/tbe/sin_ds.py

@@ -1,38 +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.
-# ============================================================================
-
-"""Sin op"""
-from mindspore.ops.op_info_register import op_info_register, TBERegOp, DataType
-
-sin_op_info = TBERegOp("Sin") \
-    .fusion_type("ELEMWISE") \
-    .async_flag(False) \
-    .binfile_name("sin.so") \
-    .compute_cost(10) \
-    .kernel_name("sin") \
-    .partial_flag(True) \
-    .dynamic_shape(True) \
-    .op_pattern("formatAgnostic") \
-    .input(0, "x", False, "required", "all") \
-    .output(0, "y", False, "required", "all") \
-    .dtype_format(DataType.F16_None, DataType.F16_None) \
-    .dtype_format(DataType.F32_None, DataType.F32_None) \
-    .get_op_info()
-
-
-@op_info_register(sin_op_info)
-def _sin_ds_tbe():
-    """Sin TBE register"""
-    return

mindspore/ops/_op_impl/tbe/sinh.py

@@ -1,37 +0,0 @@
-# Copyright 2020 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.
-# ============================================================================
-
-"""Sinh op"""
-from mindspore.ops.op_info_register import op_info_register, TBERegOp, DataType
-
-sinh_op_info = TBERegOp("Sinh") \
-    .fusion_type("ELEMWISE") \
-    .async_flag(False) \
-    .binfile_name("sinh.so") \
-    .compute_cost(10) \
-    .kernel_name("sinh") \
-    .partial_flag(True) \
-    .op_pattern("formatAgnostic") \
-    .input(0, "x", False, "required", "all") \
-    .output(0, "y", True, "required", "all") \
-    .dtype_format(DataType.F16_None, DataType.F16_None) \
-    .dtype_format(DataType.F32_None, DataType.F32_None) \
-    .get_op_info()
-
-
-@op_info_register(sinh_op_info)
-def _sinh_tbe():
-    """Sinh TBE register"""
-    return

mindspore/ops/_op_impl/tbe/sinh_ds.py

@@ -1,38 +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.
-# ============================================================================
-
-"""Sinh op"""
-from mindspore.ops.op_info_register import op_info_register, TBERegOp, DataType
-
-sinh_op_info = TBERegOp("Sinh") \
-    .fusion_type("ELEMWISE") \
-    .async_flag(False) \
-    .binfile_name("sinh.so") \
-    .compute_cost(10) \
-    .kernel_name("sinh") \
-    .partial_flag(True) \
-    .dynamic_shape(True) \
-    .op_pattern("formatAgnostic") \
-    .input(0, "x", False, "required", "all") \
-    .output(0, "y", True, "required", "all") \
-    .dtype_format(DataType.F16_None, DataType.F16_None) \
-    .dtype_format(DataType.F32_None, DataType.F32_None) \
-    .get_op_info()
-
-
-@op_info_register(sinh_op_info)
-def _sinh_ds_tbe():
-    """Sinh TBE register"""
-    return

mindspore/ops/_op_impl/tbe/slice.py

@@ -1,58 +0,0 @@
-# Copyright 2020 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.
-# ============================================================================
-
-"""Slice op"""
-from mindspore.ops.op_info_register import op_info_register, TBERegOp, DataType
-
-slice_op_info = TBERegOp("Slice") \
-    .fusion_type("OPAQUE") \
-    .async_flag(False) \
-    .binfile_name("slice.so") \
-    .compute_cost(10) \
-    .kernel_name("slice") \
-    .partial_flag(True) \
-    .dynamic_compile_static(True) \
-    .dynamic_shape(True) \
-    .input(0, "x", False, "required", "all") \
-    .input(1, "begin", False, "required", "all", "optional") \
-    .input(2, "size", False, "required", "all", "optional") \
-    .output(0, "y", False, "required", "all") \
-    .dtype_format(DataType.I8_Default, DataType.I32_Default, DataType.I32_Default, DataType.I8_Default) \
-    .dtype_format(DataType.U8_Default, DataType.I32_Default, DataType.I32_Default, DataType.U8_Default) \
-    .dtype_format(DataType.I16_Default, DataType.I32_Default, DataType.I32_Default, DataType.I16_Default) \
-    .dtype_format(DataType.U16_Default, DataType.I32_Default, DataType.I32_Default, DataType.U16_Default) \
-    .dtype_format(DataType.I32_Default, DataType.I32_Default, DataType.I32_Default, DataType.I32_Default) \
-    .dtype_format(DataType.I64_Default, DataType.I32_Default, DataType.I32_Default, DataType.I64_Default) \
-    .dtype_format(DataType.U32_Default, DataType.I32_Default, DataType.I32_Default, DataType.U32_Default) \
-    .dtype_format(DataType.U64_Default, DataType.I32_Default, DataType.I32_Default, DataType.U64_Default) \
-    .dtype_format(DataType.F16_Default, DataType.I32_Default, DataType.I32_Default, DataType.F16_Default) \
-    .dtype_format(DataType.F32_Default, DataType.I32_Default, DataType.I32_Default, DataType.F32_Default) \
-    .dtype_format(DataType.I8_Default, DataType.I64_Default, DataType.I64_Default, DataType.I8_Default) \
-    .dtype_format(DataType.U8_Default, DataType.I64_Default, DataType.I64_Default, DataType.U8_Default) \
-    .dtype_format(DataType.I16_Default, DataType.I64_Default, DataType.I64_Default, DataType.I16_Default) \
-    .dtype_format(DataType.U16_Default, DataType.I64_Default, DataType.I64_Default, DataType.U16_Default) \
-    .dtype_format(DataType.I32_Default, DataType.I64_Default, DataType.I64_Default, DataType.I32_Default) \
-    .dtype_format(DataType.I64_Default, DataType.I64_Default, DataType.I64_Default, DataType.I64_Default) \
-    .dtype_format(DataType.U32_Default, DataType.I64_Default, DataType.I64_Default, DataType.U32_Default) \
-    .dtype_format(DataType.U64_Default, DataType.I64_Default, DataType.I64_Default, DataType.U64_Default) \
-    .dtype_format(DataType.F16_Default, DataType.I64_Default, DataType.I64_Default, DataType.F16_Default) \
-    .dtype_format(DataType.F32_Default, DataType.I64_Default, DataType.I64_Default, DataType.F32_Default) \
-    .get_op_info()
-
-
-@op_info_register(slice_op_info)
-def _slice_tbe():
-    """Slice TBE register"""
-    return

mindspore/ops/_op_impl/tbe/smooth_l1_loss.py

@@ -1,45 +0,0 @@
-# Copyright 2020 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.
-# ============================================================================
-
-"""SmoothL1Loss op"""
-from mindspore.ops.op_info_register import op_info_register, TBERegOp, DataType
-
-smooth_l1_loss_op_info = TBERegOp("SmoothL1Loss") \
-    .fusion_type("OPAQUE") \
-    .async_flag(False) \
-    .binfile_name("smooth_l1_loss_v2.so") \
-    .compute_cost(10) \
-    .kernel_name("smooth_l1_loss_v2") \
-    .partial_flag(True) \
-    .attr("beta", "optional", "float", "all") \
-    .attr("reduction", "optional", "str", "all") \
-    .input(0, "predict", False, "required", "all") \
-    .input(1, "label", False, "required", "all") \
-    .output(0, "loss", False, "required", "all") \
-    .dtype_format(DataType.F16_Default, DataType.F16_Default, DataType.F16_Default) \
-    .dtype_format(DataType.F16_5HD, DataType.F16_5HD, DataType.F16_5HD) \
-    .dtype_format(DataType.F16_FracZ, DataType.F16_FracZ, DataType.F16_FracZ) \
-    .dtype_format(DataType.F16_C1HWNCoC0, DataType.F16_C1HWNCoC0, DataType.F16_C1HWNCoC0) \
-    .dtype_format(DataType.F32_Default, DataType.F32_Default, DataType.F32_Default) \
-    .dtype_format(DataType.F32_5HD, DataType.F32_5HD, DataType.F32_5HD) \
-    .dtype_format(DataType.F32_FracZ, DataType.F32_FracZ, DataType.F32_FracZ) \
-    .dtype_format(DataType.F32_C1HWNCoC0, DataType.F32_C1HWNCoC0, DataType.F32_C1HWNCoC0) \
-    .get_op_info()
-
-
-@op_info_register(smooth_l1_loss_op_info)
-def _smooth_l1_loss_tbe():
-    """SmoothL1Loss TBE register"""
-    return

mindspore/ops/_op_impl/tbe/smooth_l1_loss_ds.py

@@ -1,46 +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.
-# ============================================================================
-
-"""SmoothL1Loss op"""
-from mindspore.ops.op_info_register import op_info_register, TBERegOp, DataType
-
-smooth_l1_loss_op_info = TBERegOp("SmoothL1Loss") \
-    .fusion_type("OPAQUE") \
-    .async_flag(False) \
-    .binfile_name("smooth_l1_loss_v2.so") \
-    .compute_cost(10) \
-    .kernel_name("smooth_l1_loss_v2") \
-    .partial_flag(True) \
-    .dynamic_shape(True) \
-    .attr("beta", "optional", "float", "all") \
-    .attr("reduction", "optional", "str", "all") \
-    .input(0, "predict", False, "required", "all") \
-    .input(1, "label", False, "required", "all") \
-    .output(0, "loss", False, "required", "all") \
-    .dtype_format(DataType.F16_Default, DataType.F16_Default, DataType.F16_Default) \
-    .dtype_format(DataType.F16_5HD, DataType.F16_5HD, DataType.F16_5HD) \
-    .dtype_format(DataType.F16_FracZ, DataType.F16_FracZ, DataType.F16_FracZ) \
-    .dtype_format(DataType.F16_C1HWNCoC0, DataType.F16_C1HWNCoC0, DataType.F16_C1HWNCoC0) \
-    .dtype_format(DataType.F32_Default, DataType.F32_Default, DataType.F32_Default) \
-    .dtype_format(DataType.F32_5HD, DataType.F32_5HD, DataType.F32_5HD) \
-    .dtype_format(DataType.F32_FracZ, DataType.F32_FracZ, DataType.F32_FracZ) \
-    .dtype_format(DataType.F32_C1HWNCoC0, DataType.F32_C1HWNCoC0, DataType.F32_C1HWNCoC0) \
-    .get_op_info()
-
-
-@op_info_register(smooth_l1_loss_op_info)
-def _smooth_l1_loss_ds_tbe():
-    """SmoothL1Loss TBE register"""
-    return

mindspore/ops/_op_impl/tbe/smooth_l1_loss_grad.py

@@ -1,46 +0,0 @@
-# Copyright 2020 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.
-# ============================================================================
-
-"""SmoothL1LossGrad op"""
-from mindspore.ops.op_info_register import op_info_register, TBERegOp, DataType
-
-smooth_l1_loss_grad_op_info = TBERegOp("SmoothL1LossGrad") \
-    .fusion_type("OPAQUE") \
-    .async_flag(False) \
-    .binfile_name("smooth_l1_loss_grad_v2.so") \
-    .compute_cost(10) \
-    .kernel_name("smooth_l1_loss_grad_v2") \
-    .partial_flag(True) \
-    .attr("beta", "optional", "float", "all") \
-    .attr("reduction", "optional", "str", "all") \
-    .input(0, "predict", False, "required", "all") \
-    .input(1, "label", False, "required", "all") \
-    .input(2, "dout", False, "required", "all") \
-    .output(0, "loss", False, "required", "all") \
-    .dtype_format(DataType.F16_Default, DataType.F16_Default, DataType.F16_Default, DataType.F16_Default) \
-    .dtype_format(DataType.F16_5HD, DataType.F16_5HD, DataType.F16_5HD, DataType.F16_5HD) \
-    .dtype_format(DataType.F16_FracZ, DataType.F16_FracZ, DataType.F16_FracZ, DataType.F16_FracZ) \
-    .dtype_format(DataType.F16_C1HWNCoC0, DataType.F16_C1HWNCoC0, DataType.F16_C1HWNCoC0, DataType.F16_C1HWNCoC0) \
-    .dtype_format(DataType.F32_Default, DataType.F32_Default, DataType.F32_Default, DataType.F32_Default) \
-    .dtype_format(DataType.F32_5HD, DataType.F32_5HD, DataType.F32_5HD, DataType.F32_5HD) \
-    .dtype_format(DataType.F32_FracZ, DataType.F32_FracZ, DataType.F32_FracZ, DataType.F32_FracZ) \
-    .dtype_format(DataType.F32_C1HWNCoC0, DataType.F32_C1HWNCoC0, DataType.F32_C1HWNCoC0, DataType.F32_C1HWNCoC0) \
-    .get_op_info()
-
-
-@op_info_register(smooth_l1_loss_grad_op_info)
-def _smooth_l1_loss_grad_tbe():
-    """SmoothL1LossGrad TBE register"""
-    return

mindspore/ops/_op_impl/tbe/smooth_l1_loss_grad_ds.py

@@ -1,47 +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.
-# ============================================================================
-
-"""SmoothL1LossGrad op"""
-from mindspore.ops.op_info_register import op_info_register, TBERegOp, DataType
-
-smooth_l1_loss_grad_op_info = TBERegOp("SmoothL1LossGrad") \
-    .fusion_type("OPAQUE") \
-    .async_flag(False) \
-    .binfile_name("smooth_l1_loss_grad_v2.so") \
-    .compute_cost(10) \
-    .kernel_name("smooth_l1_loss_grad_v2") \
-    .partial_flag(True) \
-    .dynamic_shape(True) \
-    .attr("beta", "optional", "float", "all") \
-    .attr("reduction", "optional", "str", "all") \
-    .input(0, "predict", False, "required", "all") \
-    .input(1, "label", False, "required", "all") \
-    .input(2, "dout", False, "required", "all") \
-    .output(0, "loss", False, "required", "all") \
-    .dtype_format(DataType.F16_Default, DataType.F16_Default, DataType.F16_Default, DataType.F16_Default) \
-    .dtype_format(DataType.F16_5HD, DataType.F16_5HD, DataType.F16_5HD, DataType.F16_5HD) \
-    .dtype_format(DataType.F16_FracZ, DataType.F16_FracZ, DataType.F16_FracZ, DataType.F16_FracZ) \
-    .dtype_format(DataType.F16_C1HWNCoC0, DataType.F16_C1HWNCoC0, DataType.F16_C1HWNCoC0, DataType.F16_C1HWNCoC0) \
-    .dtype_format(DataType.F32_Default, DataType.F32_Default, DataType.F32_Default, DataType.F32_Default) \
-    .dtype_format(DataType.F32_5HD, DataType.F32_5HD, DataType.F32_5HD, DataType.F32_5HD) \
-    .dtype_format(DataType.F32_FracZ, DataType.F32_FracZ, DataType.F32_FracZ, DataType.F32_FracZ) \
-    .dtype_format(DataType.F32_C1HWNCoC0, DataType.F32_C1HWNCoC0, DataType.F32_C1HWNCoC0, DataType.F32_C1HWNCoC0) \
-    .get_op_info()
-
-
-@op_info_register(smooth_l1_loss_grad_op_info)
-def _smooth_l1_loss_grad_ds_tbe():
-    """SmoothL1LossGrad TBE register"""
-    return

mindspore/ops/_op_impl/tbe/soft_margin_loss.py

@@ -1,38 +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.
-# ============================================================================
-
-"""SoftMarginLoss op"""
-from mindspore.ops.op_info_register import op_info_register, TBERegOp, DataType
-
-soft_margin_loss_op_info = TBERegOp("SoftMarginLoss") \
-    .fusion_type("OPAQUE") \
-    .async_flag(False) \
-    .binfile_name("soft_margin_loss.so") \
-    .compute_cost(10) \
-    .kernel_name("soft_margin_loss") \
-    .partial_flag(True) \
-    .attr("reduction", "optional", "str", "mean,none,sum", "mean") \
-    .input(0, "input_x", False, "required", "all") \
-    .input(1, "input_y", False, "required", "all") \
-    .output(0, "output_z", False, "required", "all") \
-    .dtype_format(DataType.F32_Default, DataType.F32_Default, DataType.F32_Default) \
-    .dtype_format(DataType.F16_Default, DataType.F16_Default, DataType.F16_Default) \
-    .get_op_info()
-
-
-@op_info_register(soft_margin_loss_op_info)
-def _soft_margin_loss_tbe():
-    """SoftMarginLoss TBE register"""
-    return