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

mindspore/ops/_op_impl/tbe/inv_grad_ds.py

@@ -1,41 +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.
-# ============================================================================
-
-"""InvGrad op"""
-from mindspore.ops.op_info_register import op_info_register, TBERegOp, DataType
-
-inv_grad_ds_op_info = TBERegOp("InvGrad") \
-    .fusion_type("ELEMWISE") \
-    .async_flag(False) \
-    .binfile_name("inv_grad.so") \
-    .compute_cost(10) \
-    .kernel_name("inv_grad") \
-    .partial_flag(True) \
-    .dynamic_shape(True) \
-    .input(0, "x", False, "required", "all") \
-    .input(1, "grad", False, "required", "all") \
-    .output(0, "y", False, "required", "all") \
-    .op_pattern("broadcast") \
-    .dtype_format(DataType.F16_None, DataType.F16_None, DataType.F16_None) \
-    .dtype_format(DataType.F32_None, DataType.F32_None, DataType.F32_None) \
-    .dtype_format(DataType.I32_None, DataType.I32_None, DataType.I32_None) \
-    .dtype_format(DataType.I8_None, DataType.I8_None, DataType.I8_None) \
-    .get_op_info()
-
-
-@op_info_register(inv_grad_ds_op_info)
-def _inv_grad_ds_tbe():
-    """InvGrad TBE register"""
-    return

mindspore/ops/_op_impl/tbe/invert.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.
-# ============================================================================
-
-"""Invert op"""
-from mindspore.ops.op_info_register import op_info_register, TBERegOp, DataType
-
-invert_op_info = TBERegOp("Invert") \
-    .fusion_type("OPAQUE") \
-    .async_flag(False) \
-    .binfile_name("invert.so") \
-    .compute_cost(10) \
-    .kernel_name("invert") \
-    .partial_flag(True) \
-    .input(0, "x", False, "required", "all") \
-    .output(0, "y", False, "required", "all") \
-    .op_pattern("formatAgnostic") \
-    .dtype_format(DataType.I16_None, DataType.I16_None) \
-    .dtype_format(DataType.U16_None, DataType.U16_None) \
-    .get_op_info()
-
-
-@op_info_register(invert_op_info)
-def _invert_tbe():
-    """Invert TBE register"""
-    return

mindspore/ops/_op_impl/tbe/invert_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.
-# ============================================================================
-
-"""Invert op"""
-from mindspore.ops.op_info_register import op_info_register, TBERegOp, DataType
-
-invert_ds_op_info = TBERegOp("Invert") \
-    .fusion_type("OPAQUE") \
-    .async_flag(False) \
-    .binfile_name("invert.so") \
-    .compute_cost(10) \
-    .kernel_name("invert") \
-    .partial_flag(True) \
-    .dynamic_shape(True) \
-    .input(0, "x", False, "required", "all") \
-    .output(0, "y", False, "required", "all") \
-    .op_pattern("formatAgnostic") \
-    .dtype_format(DataType.I16_None, DataType.I16_None) \
-    .dtype_format(DataType.U16_None, DataType.U16_None) \
-    .get_op_info()
-
-
-@op_info_register(invert_ds_op_info)
-def _invert_ds_tbe():
-    """Invert TBE register"""
-    return

mindspore/ops/_op_impl/tbe/iou.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.
-# ============================================================================
-
-"""Iou op"""
-from mindspore.ops.op_info_register import op_info_register, TBERegOp, DataType
-
-iou_op_info = TBERegOp("IOU") \
-    .fusion_type("OPAQUE") \
-    .async_flag(False) \
-    .binfile_name("iou.so") \
-    .compute_cost(10) \
-    .kernel_name("iou") \
-    .partial_flag(True) \
-    .attr("mode", "optional", "str", "all", "iou") \
-    .attr("eps", "optional", "float", "all", "1.0") \
-    .input(0, "bboxes", False, "required", "all") \
-    .input(1, "gtboxes", False, "required", "all") \
-    .output(0, "overlap", False, "required", "all") \
-    .dtype_format(DataType.F16_Default, DataType.F16_Default, DataType.F16_Default) \
-    .get_op_info()
-
-
-@op_info_register(iou_op_info)
-def _iou_tbe():
-    """Iou TBE register"""
-    return

mindspore/ops/_op_impl/tbe/iou_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.
-# ============================================================================
-
-"""Iou op"""
-from mindspore.ops.op_info_register import op_info_register, TBERegOp, DataType
-
-iou_op_info = TBERegOp("IOU") \
-    .fusion_type("OPAQUE") \
-    .async_flag(False) \
-    .binfile_name("iou.so") \
-    .compute_cost(10) \
-    .kernel_name("iou") \
-    .partial_flag(True) \
-    .dynamic_shape(True)\
-    .attr("mode", "optional", "str", "all", "iou") \
-    .attr("eps", "optional", "float", "all", "1.0") \
-    .input(0, "bboxes", False, "required", "all") \
-    .input(1, "gtboxes", False, "required", "all") \
-    .output(0, "overlap", False, "required", "all") \
-    .dtype_format(DataType.F16_Default, DataType.F16_Default, DataType.F16_Default) \
-    .get_op_info()
-
-
-@op_info_register(iou_op_info)
-def _iou_ds_tbe():
-    """Iou TBE register"""
-    return

mindspore/ops/_op_impl/tbe/is_close.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.
-# ============================================================================
-
-"""IsClose op"""
-from mindspore.ops.op_info_register import op_info_register, TBERegOp, DataType
-
-is_close_op_info = TBERegOp("IsClose") \
-    .fusion_type("ELEMWISE") \
-    .async_flag(False) \
-    .binfile_name("is_close.so") \
-    .compute_cost(10) \
-    .kernel_name("is_close") \
-    .partial_flag(True) \
-    .attr("rtol", "optional", "float", "all", "1e-05")\
-    .attr("atol", "optional", "float", "all", "1e-08")\
-    .attr("equal_nan", "optional", "bool", "true,false", "False")\
-    .input(0, "x1", False, "required", "all") \
-    .input(1, "x2", False, "required", "all") \
-    .output(0, "y", False, "required", "all") \
-    .dtype_format(DataType.F16_Default, DataType.F16_Default, DataType.BOOL_Default) \
-    .dtype_format(DataType.F32_Default, DataType.F32_Default, DataType.BOOL_Default) \
-    .dtype_format(DataType.I32_Default, DataType.I32_Default, DataType.BOOL_Default) \
-    .get_op_info()
-
-@op_info_register(is_close_op_info)
-def _is_close_tbe():
-    """IsClose TBE register"""
-    return

mindspore/ops/_op_impl/tbe/kl_div_loss.py

@@ -1,38 +0,0 @@
-# Copyright 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.
-# ============================================================================
-
-"""KLDivLoss op"""
-from mindspore.ops.op_info_register import op_info_register, TBERegOp, DataType
-
-kl_div_loss_op_info = TBERegOp("KLDivLoss") \
-    .fusion_type("OPAQUE") \
-    .async_flag(False) \
-    .binfile_name("kl_div.so") \
-    .compute_cost(10) \
-    .kernel_name("kl_div") \
-    .partial_flag(True) \
-    .attr("reduction", "required", "str", "all") \
-    .input(0, "x", False, "required", "all") \
-    .input(1, "target", False, "required", "all") \
-    .output(0, "y", False, "required", "all") \
-    .is_dynamic_format(True) \
-    .dtype_format(DataType.None_None, DataType.None_None, DataType.None_None) \
-    .get_op_info()
-
-
-@op_info_register(kl_div_loss_op_info)
-def _kl_div_loss_tbe():
-    """KLDivLoss TBE register"""
-    return

mindspore/ops/_op_impl/tbe/kl_div_loss_ds.py

@@ -1,39 +0,0 @@
-# Copyright 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.
-# ============================================================================
-
-"""KLDivLoss op"""
-from mindspore.ops.op_info_register import op_info_register, TBERegOp, DataType
-
-kl_div_loss_ds_op_info = TBERegOp("KLDivLoss") \
-    .fusion_type("OPAQUE") \
-    .async_flag(False) \
-    .binfile_name("kl_div.so") \
-    .compute_cost(10) \
-    .kernel_name("kl_div") \
-    .partial_flag(True) \
-    .dynamic_shape(True) \
-    .attr("reduction", "required", "str", "all") \
-    .input(0, "x", False, "required", "all") \
-    .input(1, "target", False, "required", "all") \
-    .output(0, "y", False, "required", "all") \
-    .is_dynamic_format(True) \
-    .dtype_format(DataType.None_None, DataType.None_None, DataType.None_None) \
-    .get_op_info()
-
-
-@op_info_register(kl_div_loss_ds_op_info)
-def _kl_div_loss_ds_tbe():
-    """KLDivLoss TBE register"""
-    return

mindspore/ops/_op_impl/tbe/kl_div_loss_grad.py

@@ -1,40 +0,0 @@
-# Copyright 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.
-# ============================================================================
-
-"""KLDivLossGrad op"""
-from mindspore.ops.op_info_register import op_info_register, TBERegOp, DataType
-
-kl_div_loss_grad_op_info = TBERegOp("KLDivLossGrad") \
-    .fusion_type("OPAQUE") \
-    .async_flag(False) \
-    .binfile_name("kl_div_loss_grad.so") \
-    .compute_cost(10) \
-    .kernel_name("kl_div_loss_grad") \
-    .partial_flag(True) \
-    .attr("reduction", "optional", "str", "all", "mean") \
-    .attr("log_target", "optional", "bool", "all", "false") \
-    .input(0, "grad", False, "required", "all") \
-    .input(1, "input", False, "required", "all") \
-    .input(2, "target", False, "required", "all") \
-    .output(0, "y", False, "required", "all") \
-    .dtype_format(DataType.F32_Default, DataType.F32_Default, DataType.F32_Default, DataType.F32_Default) \
-    .dtype_format(DataType.F16_Default, DataType.F16_Default, DataType.F16_Default, DataType.F16_Default) \
-    .get_op_info()
-
-
-@op_info_register(kl_div_loss_grad_op_info)
-def _kl_div_loss_grad_tbe():
-    """KLDivLossGrad TBE register"""
-    return

mindspore/ops/_op_impl/tbe/l2_loss.py

@@ -1,36 +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.
-# ============================================================================
-
-"""L2Loss op"""
-from mindspore.ops.op_info_register import op_info_register, TBERegOp, DataType
-
-l2_loss_op_info = TBERegOp("L2Loss") \
-    .fusion_type("OPAQUE") \
-    .async_flag(False) \
-    .binfile_name("l2_loss.so") \
-    .compute_cost(10) \
-    .kernel_name("l2_loss") \
-    .partial_flag(True) \
-    .input(0, "x", None, "required", None) \
-    .output(0, "y", True, "required", "all") \
-    .is_dynamic_format(True) \
-    .dtype_format(DataType.None_None, DataType.None_None) \
-    .get_op_info()
-
-
-@op_info_register(l2_loss_op_info)
-def _l2_loss_tbe():
-    """L2Loss TBE register"""
-    return

mindspore/ops/_op_impl/tbe/l2_loss_ds.py

@@ -1,37 +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.
-# ============================================================================
-
-"""L2Loss op"""
-from mindspore.ops.op_info_register import op_info_register, TBERegOp, DataType
-
-l2_loss_op_info = TBERegOp("L2Loss") \
-    .fusion_type("OPAQUE") \
-    .async_flag(False) \
-    .binfile_name("l2_loss.so") \
-    .compute_cost(10) \
-    .kernel_name("l2_loss") \
-    .partial_flag(True) \
-    .dynamic_shape(True) \
-    .input(0, "x", None, "required", None) \
-    .output(0, "y", True, "required", "all") \
-    .is_dynamic_format(True) \
-    .dtype_format(DataType.None_None, DataType.None_None) \
-    .get_op_info()
-
-
-@op_info_register(l2_loss_op_info)
-def _l2_loss_ds_tbe():
-    """L2Loss TBE register"""
-    return

mindspore/ops/_op_impl/tbe/l2_normalize.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.
-# ============================================================================
-
-"""L2Normalize op"""
-from mindspore.ops.op_info_register import op_info_register, TBERegOp, DataType
-
-l2_normalize_op_info = TBERegOp("L2Normalize") \
-    .fusion_type("OPAQUE") \
-    .async_flag(False) \
-    .binfile_name("l2_normalize.so") \
-    .compute_cost(10) \
-    .kernel_name("l2_normalize") \
-    .partial_flag(True) \
-    .attr("axis", "required", "listInt", "all") \
-    .attr("epsilon", "required", "float", "all") \
-    .input(0, "x", False, "required", "all") \
-    .output(0, "y", True, "required", "all") \
-    .dtype_format(DataType.F16_Default, DataType.F16_Default) \
-    .dtype_format(DataType.F32_Default, DataType.F32_Default) \
-    .get_op_info()
-
-
-@op_info_register(l2_normalize_op_info)
-def _l2_normalize_tbe():
-    """L2Normalize TBE register"""
-    return

mindspore/ops/_op_impl/tbe/l2_normalize_grad.py

@@ -1,40 +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.
-# ============================================================================
-
-"""L2NormalizeGrad op"""
-from mindspore.ops.op_info_register import op_info_register, TBERegOp, DataType
-
-l2_normalize_grad_op_info = TBERegOp("L2NormalizeGrad") \
-    .fusion_type("OPAQUE") \
-    .async_flag(False) \
-    .binfile_name("l2_normalize_grad.so") \
-    .compute_cost(10) \
-    .kernel_name("l2_normalize_grad") \
-    .partial_flag(True) \
-    .attr("axis", "required", "listInt", "all") \
-    .attr("epsilon", "required", "float", "all") \
-    .input(0, "x", False, "required", "all") \
-    .input(1, "y", False, "required", "all") \
-    .input(2, "dy", False, "required", "all") \
-    .output(0, "dx", True, "required", "all") \
-    .dtype_format(DataType.F16_Default, DataType.F16_Default, DataType.F16_Default, DataType.F16_Default) \
-    .dtype_format(DataType.F32_Default, DataType.F32_Default, DataType.F32_Default, DataType.F32_Default) \
-    .get_op_info()
-
-
-@op_info_register(l2_normalize_grad_op_info)
-def _l2_normalize_grad_tbe():
-    """L2NormalizeGrad TBE register"""
-    return

mindspore/ops/_op_impl/tbe/lamb_apply_optimizer_assign.py

@@ -1,55 +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.
-# ============================================================================
-
-"""LambApplyOptimizerAssign op"""
-from mindspore.ops.op_info_register import op_info_register, TBERegOp, DataType
-
-lamb_apply_optimizer_assign_op_info = TBERegOp("LambApplyOptimizerAssign") \
-    .fusion_type("ELEMWISE") \
-    .async_flag(False) \
-    .binfile_name("lamb_apply_optimizer_assign.so") \
-    .compute_cost(10) \
-    .kernel_name("lamb_apply_optimizer_assign") \
-    .partial_flag(True) \
-    .input(0, "grad", False, "required", "all") \
-    .input(1, "inputv", False, "required", "all") \
-    .input(2, "inputm", False, "required", "all") \
-    .input(3, "input3", False, "required", "all") \
-    .input(4, "mul0_x", False, "required", "all") \
-    .input(5, "mul1_x", False, "required", "all") \
-    .input(6, "mul2_x", False, "required", "all") \
-    .input(7, "mul3_x", False, "required", "all") \
-    .input(8, "add2_y", False, "required", "all") \
-    .input(9, "steps", False, "required", "all") \
-    .input(10, "do_use_weight", False, "required", "all") \
-    .input(11, "weight_decay_rate", False, "required", "all") \
-    .output(0, "output0", False, "required", "all") \
-    .output(0, "inputv", False, "required", "all") \
-    .output(0, "inputm", False, "required", "all") \
-    .dtype_format(DataType.F16_Default, DataType.F16_Default, DataType.F16_Default, DataType.F16_Default,
-                  DataType.F16_Default, DataType.F16_Default, DataType.F16_Default, DataType.F16_Default,
-                  DataType.F16_Default, DataType.F16_Default, DataType.F16_Default, DataType.F16_Default,
-                  DataType.F16_Default, DataType.F16_Default, DataType.F16_Default) \
-    .dtype_format(DataType.F32_Default, DataType.F32_Default, DataType.F32_Default, DataType.F32_Default,
-                  DataType.F32_Default, DataType.F32_Default, DataType.F32_Default, DataType.F32_Default,
-                  DataType.F32_Default, DataType.F32_Default, DataType.F32_Default, DataType.F32_Default,
-                  DataType.F32_Default, DataType.F32_Default, DataType.F32_Default) \
-    .get_op_info()
-
-
-@op_info_register(lamb_apply_optimizer_assign_op_info)
-def _lamb_apply_optimizer_assign_tbe():
-    """LambApplyOptimizerAssign TBE register"""
-    return

mindspore/ops/_op_impl/tbe/lamb_apply_weight_assign.py

@@ -1,42 +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.
-# ============================================================================
-
-"""LambApplyWeightAssign op"""
-from mindspore.ops.op_info_register import op_info_register, TBERegOp, DataType
-
-lamb_apply_weight_assign_op_info = TBERegOp("LambApplyWeightAssign") \
-    .fusion_type("ELEMWISE") \
-    .async_flag(False) \
-    .binfile_name("lamb_apply_weight_assign.so") \
-    .compute_cost(10) \
-    .kernel_name("lamb_apply_weight_assign") \
-    .partial_flag(True) \
-    .input(0, "input0", False, "required", "all") \
-    .input(1, "input1", False, "required", "all") \
-    .input(2, "input2", False, "required", "all") \
-    .input(3, "input3", False, "required", "all") \
-    .input(4, "input_param", False, "required", "all") \
-    .output(0, "input_param", False, "required", "all") \
-    .dtype_format(DataType.F16_Default, DataType.F16_Default, DataType.F16_Default, DataType.F16_Default,
-                  DataType.F16_Default, DataType.F16_Default) \
-    .dtype_format(DataType.F32_Default, DataType.F32_Default, DataType.F32_Default, DataType.F32_Default,
-                  DataType.F32_Default, DataType.F32_Default) \
-    .get_op_info()
-
-
-@op_info_register(lamb_apply_weight_assign_op_info)
-def _lamb_apply_weight_assign_tbe():
-    """LambApplyWeightAssign TBE register"""
-    return