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

mindspore/ops/_op_impl/tbe/atan2_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.
-# ============================================================================
-
-"""Atan2 op"""
-from mindspore.ops.op_info_register import op_info_register, TBERegOp, DataType
-
-atan2_ds_op_info = TBERegOp("Atan2") \
-    .fusion_type("ELEMWISE") \
-    .async_flag(False) \
-    .binfile_name("atan2.so") \
-    .compute_cost(10) \
-    .kernel_name("atan2") \
-    .partial_flag(True) \
-    .op_pattern("broadcast") \
-    .dynamic_shape(True) \
-    .input(0, "x1", False, "required", "all") \
-    .input(1, "x2", False, "required", "all") \
-    .output(0, "y", False, "required", "all") \
-    .dtype_format(DataType.F16_None, DataType.F16_None, DataType.F16_None) \
-    .dtype_format(DataType.F32_None, DataType.F32_None, DataType.F32_None) \
-    .get_op_info()
-
-
-@op_info_register(atan2_ds_op_info)
-def _atan2_ds_tbe():
-    """Atan2 TBE register"""
-    return

mindspore/ops/_op_impl/tbe/atan_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.
-# ============================================================================
-
-"""Atan op"""
-from mindspore.ops.op_info_register import op_info_register, TBERegOp, DataType
-
-atan_op_info = TBERegOp("Atan") \
-    .fusion_type("ELEMWISE") \
-    .async_flag(False) \
-    .binfile_name("atan.so") \
-    .compute_cost(10) \
-    .kernel_name("atan") \
-    .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(atan_op_info)
-def _atan_ds_tbe():
-    """Atan TBE register"""
-    return

mindspore/ops/_op_impl/tbe/atan_grad.py

@@ -1,43 +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.
-# ============================================================================
-
-"""AtanGrad op"""
-from mindspore.ops.op_info_register import op_info_register, TBERegOp, DataType
-
-atan_grad_op_info = TBERegOp("AtanGrad") \
-    .fusion_type("ELEMWISE") \
-    .async_flag(False) \
-    .binfile_name("atan_grad.so") \
-    .compute_cost(10) \
-    .kernel_name("atan_grad") \
-    .partial_flag(True) \
-    .input(0, "y", False, "required", "all") \
-    .input(1, "dy", False, "required", "all") \
-    .output(0, "z", False, "required", "all") \
-    .dtype_format(DataType.F16_5HD, DataType.F16_5HD, DataType.F16_5HD) \
-    .dtype_format(DataType.F16_FracZ, DataType.F16_FracNZ, DataType.F16_FracZ) \
-    .dtype_format(DataType.F16_C1HWNCoC0, DataType.F16_C1HWNCoC0, DataType.F16_C1HWNCoC0) \
-    .dtype_format(DataType.F16_Default, DataType.F16_Default, DataType.F16_Default) \
-    .dtype_format(DataType.F32_5HD, DataType.F32_5HD, DataType.F32_5HD) \
-    .dtype_format(DataType.F32_FracZ, DataType.F32_FracNZ, DataType.F32_FracZ) \
-    .dtype_format(DataType.F32_C1HWNCoC0, DataType.F32_C1HWNCoC0, DataType.F32_C1HWNCoC0) \
-    .dtype_format(DataType.F32_Default, DataType.F32_Default, DataType.F32_Default) \
-    .get_op_info()
-
-
-@op_info_register(atan_grad_op_info)
-def _atan_grad_tbe():
-    """AtanGrad TBE register"""
-    return

mindspore/ops/_op_impl/tbe/atan_grad_ds.py

@@ -1,44 +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.
-# ============================================================================
-
-"""AtanGrad op"""
-from mindspore.ops.op_info_register import op_info_register, TBERegOp, DataType
-
-atan_grad_op_info = TBERegOp("AtanGrad") \
-    .fusion_type("ELEMWISE") \
-    .async_flag(False) \
-    .binfile_name("atan_grad.so") \
-    .compute_cost(10) \
-    .kernel_name("atan_grad") \
-    .partial_flag(True) \
-    .dynamic_shape(True) \
-    .input(0, "y", False, "required", "all") \
-    .input(1, "dy", False, "required", "all") \
-    .output(0, "z", False, "required", "all") \
-    .dtype_format(DataType.F16_5HD, DataType.F16_5HD, DataType.F16_5HD) \
-    .dtype_format(DataType.F16_FracZ, DataType.F16_FracNZ, DataType.F16_FracZ) \
-    .dtype_format(DataType.F16_C1HWNCoC0, DataType.F16_C1HWNCoC0, DataType.F16_C1HWNCoC0) \
-    .dtype_format(DataType.F16_Default, DataType.F16_Default, DataType.F16_Default) \
-    .dtype_format(DataType.F32_5HD, DataType.F32_5HD, DataType.F32_5HD) \
-    .dtype_format(DataType.F32_FracZ, DataType.F32_FracNZ, DataType.F32_FracZ) \
-    .dtype_format(DataType.F32_C1HWNCoC0, DataType.F32_C1HWNCoC0, DataType.F32_C1HWNCoC0) \
-    .dtype_format(DataType.F32_Default, DataType.F32_Default, DataType.F32_Default) \
-    .get_op_info()
-
-
-@op_info_register(atan_grad_op_info)
-def _atan_grad_ds_tbe():
-    """AtanGrad TBE register"""
-    return

mindspore/ops/_op_impl/tbe/atanh.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.
-# ============================================================================
-
-"""Atanh op"""
-from mindspore.ops.op_info_register import op_info_register, TBERegOp, DataType
-
-atanh_op_info = TBERegOp("Atanh") \
-    .fusion_type("ELEMWISE") \
-    .async_flag(False) \
-    .binfile_name("atanh.so") \
-    .compute_cost(10) \
-    .kernel_name("atanh") \
-    .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(atanh_op_info)
-def _atanh_tbe():
-    """Atanh TBE register"""
-    return

mindspore/ops/_op_impl/tbe/atanh_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.
-# ============================================================================
-
-"""Atanh op"""
-from mindspore.ops.op_info_register import op_info_register, TBERegOp, DataType
-
-atanh_op_info = TBERegOp("Atanh") \
-    .fusion_type("ELEMWISE") \
-    .async_flag(False) \
-    .binfile_name("atanh.so") \
-    .compute_cost(10) \
-    .kernel_name("atanh") \
-    .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(atanh_op_info)
-def _atanh_ds_tbe():
-    """Atanh TBE register"""
-    return

mindspore/ops/_op_impl/tbe/avg_pool.py

@@ -1,43 +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.
-# ============================================================================
-
-"""AvgPool op"""
-from mindspore.ops.op_info_register import op_info_register, TBERegOp, DataType
-
-avg_pool_op_info = TBERegOp("AvgPool") \
-    .fusion_type("OPAQUE") \
-    .async_flag(False) \
-    .binfile_name("avg_pool.so") \
-    .compute_cost(10) \
-    .kernel_name("avg_pool") \
-    .partial_flag(True) \
-    .attr("kernel_size", "required", "listInt", "all", "[]") \
-    .attr("strides", "required", "listInt", "all", "[]") \
-    .attr("pad_mode", "required", "str", "all") \
-    .attr("format", "optional", "str", "all") \
-    .attr("offset_x", "optional", "int", "all", "0") \
-    .input(0, "x", False, "required", "all") \
-    .input(1, "filter", False, "optional", "all") \
-    .input(2, "bias", False, "optional", "all") \
-    .output(0, "y", False, "required", "all") \
-    .dtype_format(DataType.F16_5HD, DataType.F16_FracZ, DataType.F16_Default, DataType.F16_5HD) \
-    .dtype_format(DataType.I8_5HD, DataType.I8_C1HWNCoC0, DataType.I32_Default, DataType.I32_5HD) \
-    .get_op_info()
-
-
-@op_info_register(avg_pool_op_info)
-def _avg_pool_tbe():
-    """AvgPool TBE register"""
-    return

mindspore/ops/_op_impl/tbe/avg_pool_3d.py

@@ -1,44 +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.
-# ============================================================================
-
-"""AvgPool3D op"""
-from mindspore.ops.op_info_register import op_info_register, TBERegOp, DataType
-
-avg_pool_3d_op_info = TBERegOp("AvgPool3D") \
-    .fusion_type("OPAQUE") \
-    .async_flag(False) \
-    .binfile_name("avg_pool3d_d.so") \
-    .compute_cost(10) \
-    .kernel_name("avg_pool3d_d") \
-    .partial_flag(True) \
-    .attr("kernel_size", "required", "listInt", "all", "[]") \
-    .attr("strides", "required", "listInt", "all", "[]") \
-    .attr("pad_list", "required", "listInt", "all", "[]") \
-    .attr("ceil_mode", "optional", "bool", "all") \
-    .attr("count_include_pad", "optional", "bool", "all") \
-    .attr("divisor_override", "optional", "int", "all", '0') \
-    .attr("format", "optional", "str", "all", 'NCDHW') \
-    .input(0, "x", False, "required", "all") \
-    .input(1, "filter", False, "optional", "all") \
-    .input(2, "multiplier", False, "optional", "all") \
-    .output(0, "y", False, "required", "all") \
-    .dtype_format(DataType.F16_NDC1HWC0, DataType.F16_FRACTAL_Z_3D, DataType.F16_NDC1HWC0, DataType.F16_NDC1HWC0) \
-    .get_op_info()
-
-
-@op_info_register(avg_pool_3d_op_info)
-def _avg_pool_3d_tbe():
-    """AvgPool3D TBE register"""
-    return

mindspore/ops/_op_impl/tbe/avg_pool_3d_grad.py

@@ -1,45 +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.
-# ============================================================================
-
-"""AvgPool3DGrad op"""
-from mindspore.ops.op_info_register import op_info_register, TBERegOp, DataType
-
-avg_pool_3d_grad_op_info = TBERegOp("AvgPool3DGradD") \
-    .fusion_type("OPAQUE") \
-    .async_flag(False) \
-    .binfile_name("avg_pool3d_grad_d.so") \
-    .compute_cost(10) \
-    .kernel_name("avg_pool3d_grad_d") \
-    .partial_flag(True) \
-    .attr("origin_input_shape", "required", "listInt", "all", "[]") \
-    .attr("kernel_size", "required", "listInt", "all", "[]") \
-    .attr("strides", "required", "listInt", "all", "[]") \
-    .attr("pad_list", "required", "listInt", "all", "[]") \
-    .attr("ceil_mode", "optional", "bool", "all") \
-    .attr("count_include_pad", "optional", "bool", "all") \
-    .attr("divisor_override", "optional", "int", "all", '0') \
-    .attr("format", "optional", "str", "all", 'NCDHW') \
-    .input(0, "grads", False, "required", "all") \
-    .input(1, "filter", False, "optional", "all") \
-    .input(2, "multiplier", False, "optional", "all") \
-    .output(0, "output", False, "required", "all") \
-    .dtype_format(DataType.F16_NDC1HWC0, DataType.F16_FRACTAL_Z_3D, DataType.F16_NDC1HWC0, DataType.F16_NDC1HWC0) \
-    .get_op_info()
-
-
-@op_info_register(avg_pool_3d_grad_op_info)
-def _avg_pool_3d_grad_tbe():
-    """AvgPool3DGrad TBE register"""
-    return

mindspore/ops/_op_impl/tbe/avg_pool_ds.py

@@ -1,44 +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.
-# ============================================================================
-
-"""AvgPool op"""
-from mindspore.ops.op_info_register import op_info_register, TBERegOp, DataType
-
-avg_pool_op_info = TBERegOp("AvgPool") \
-    .fusion_type("OPAQUE") \
-    .async_flag(False) \
-    .binfile_name("avg_pool.so") \
-    .compute_cost(10) \
-    .kernel_name("avg_pool") \
-    .dynamic_shape(True) \
-    .partial_flag(True) \
-    .attr("kernel_size", "required", "listInt", "all", "[]") \
-    .attr("strides", "required", "listInt", "all", "[]") \
-    .attr("pad_mode", "required", "str", "all") \
-    .attr("format", "optional", "str", "all") \
-    .attr("offset_x", "optional", "int", "all", "0") \
-    .input(0, "x", False, "required", "all") \
-    .input(1, "filter", False, "optional", "all") \
-    .input(2, "bias", False, "optional", "all") \
-    .output(0, "y", False, "required", "all") \
-    .dtype_format(DataType.F16_5HD, DataType.F16_FracZ, DataType.F16_Default, DataType.F16_5HD) \
-    .dtype_format(DataType.I8_5HD, DataType.I8_C1HWNCoC0, DataType.I32_Default, DataType.I32_5HD) \
-    .get_op_info()
-
-
-@op_info_register(avg_pool_op_info)
-def _avg_pool_ds_tbe():
-    """AvgPool TBE register"""
-    return

mindspore/ops/_op_impl/tbe/avg_pool_grad.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.
-# ============================================================================
-
-"""AvgPoolGrad op"""
-from mindspore.ops.op_info_register import op_info_register, TBERegOp, DataType
-
-avg_pool_grad_op_info = TBERegOp("AvgPoolGrad") \
-    .fusion_type("OPAQUE") \
-    .async_flag(False) \
-    .binfile_name("avg_pool_grad_d.so") \
-    .compute_cost(10) \
-    .kernel_name("avg_pool_grad_d") \
-    .partial_flag(True) \
-    .attr("x_origin", "required", "listInt", "all", "[]") \
-    .attr("kernel_size", "required", "listInt", "all", "[]") \
-    .attr("strides", "required", "listInt", "all", "[]") \
-    .attr("pad_mode", "required", "str", "all") \
-    .attr("format", "optional", "str", "all") \
-    .input(0, "input_grad", False, "required", "all") \
-    .input(1, "mean_matrix", False, "optional", "all") \
-    .input(2, "kernel_matrix", False, "optional", "all") \
-    .output(0, "out_grad", True, "required", "all") \
-    .dtype_format(DataType.F16_5HD, DataType.F16_5HD, DataType.F16_C1HWNCoC0, DataType.F16_5HD) \
-    .get_op_info()
-
-
-@op_info_register(avg_pool_grad_op_info)
-def _avg_pool_grad_tbe():
-    """AvgPoolGrad TBE register"""
-    return

mindspore/ops/_op_impl/tbe/avg_pool_grad_vm.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.
-# ============================================================================
-
-"""AvgPoolGradVm op"""
-from mindspore.ops.op_info_register import op_info_register, TBERegOp, DataType
-
-avg_pool_grad_vm_op_info = TBERegOp("AvgPoolGradVm") \
-    .fusion_type("OPAQUE") \
-    .async_flag(False) \
-    .binfile_name("avg_pool_grad_d.so") \
-    .compute_cost(10) \
-    .kernel_name("avg_pool_grad_d") \
-    .partial_flag(True) \
-    .attr("x_origin", "required", "listInt", "all", "[]") \
-    .attr("kernel_size", "required", "listInt", "all", "[]") \
-    .attr("strides", "required", "listInt", "all", "[]") \
-    .attr("pad_mode", "required", "str", "all") \
-    .attr("format", "optional", "str", "all") \
-    .input(0, "input_grad", False, "required", "all") \
-    .input(1, "mean_matrix", False, "optional", "all") \
-    .input(2, "kernel_matrix", False, "optional", "all") \
-    .output(0, "out_grad", True, "required", "all") \
-    .dtype_format(DataType.F16_5HD, DataType.F16_5HD, DataType.F16_C1HWNCoC0, DataType.F16_5HD) \
-    .get_op_info()
-
-
-@op_info_register(avg_pool_grad_vm_op_info)
-def _avg_pool_grad_vm_tbe():
-    """AvgPoolGradVm TBE register"""
-    return

mindspore/ops/_op_impl/tbe/basic_lstm_cell.py

@@ -1,57 +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.
-# ============================================================================
-
-"""BasicLSTMCell op"""
-from mindspore.ops.op_info_register import op_info_register, TBERegOp, DataType
-
-basic_lstm_cell_op_info = TBERegOp("BasicLSTMCell") \
-    .fusion_type("OPAQUE") \
-    .async_flag(False) \
-    .binfile_name("basic_lstm_cell.so") \
-    .compute_cost(10) \
-    .kernel_name("basic_lstm_cell") \
-    .attr("keep_prob", "optional", "float", "all") \
-    .attr("forget_bias", "optional", "float", "all") \
-    .attr("state_is_tuple", "optional", "bool", "true") \
-    .attr("activation", "optional", "str", "all") \
-    .partial_flag(True) \
-    .input(0, "x", False, "required", "all") \
-    .input(1, "h", False, "required", "all") \
-    .input(2, "c", False, "required", "all") \
-    .input(3, "w", False, "required", "all", reshape_type="CN") \
-    .input(4, "b", False, "required", "all") \
-    .input(5, "mask", False, "optional", "all") \
-    .output(0, "ct", False, "required", "all") \
-    .output(1, "ht", False, "required", "all") \
-    .output(2, "it", False, "optional", "all") \
-    .output(3, "jt", False, "optional", "all") \
-    .output(4, "ft", False, "optional", "all") \
-    .output(5, "ot", False, "optional", "all") \
-    .output(6, "tanhct", False, "optional", "all") \
-    .dtype_format(DataType.F16_FracNZ, DataType.F16_FracNZ, DataType.F32_FracNZ, DataType.F16_FracZNLSTM,
-                  DataType.F32_Default, DataType.U8_Default, DataType.F32_FracNZ, DataType.F16_FracNZ,
-                  DataType.F32_FracNZ, DataType.F32_FracNZ, DataType.F32_FracNZ, DataType.F32_FracNZ,
-                  DataType.F32_FracNZ) \
-    .dtype_format(DataType.F16_FracNZ, DataType.F16_FracNZ, DataType.F16_FracNZ, DataType.F16_FracZNLSTM,
-                  DataType.F16_Default, DataType.U8_Default, DataType.F16_FracNZ, DataType.F16_FracNZ,
-                  DataType.F16_FracNZ, DataType.F16_FracNZ, DataType.F16_FracNZ, DataType.F16_FracNZ,
-                  DataType.F16_FracNZ) \
-    .get_op_info()
-
-
-@op_info_register(basic_lstm_cell_op_info)
-def _basic_lstm_cell_tbe():
-    """BasicLSTMCell TBE register"""
-    return

mindspore/ops/_op_impl/tbe/basic_lstm_cell_c_state_grad.py

@@ -1,50 +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.
-# ============================================================================
-
-"""BasicLSTMCellCStateGrad op"""
-from mindspore.ops.op_info_register import op_info_register, TBERegOp, DataType
-
-basic_lstm_cell_c_state_grad_op_info = TBERegOp("BasicLSTMCellCStateGrad") \
-    .fusion_type("OPAQUE") \
-    .async_flag(False) \
-    .binfile_name("basic_lstm_cell_c_state_grad.so") \
-    .compute_cost(10) \
-    .kernel_name("basic_lstm_cell_c_state_grad") \
-    .attr("forget_bias", "optional", "float", "all", "1") \
-    .attr("activation", "optional", "str", "all", "None") \
-    .partial_flag(True) \
-    .input(0, "c", False, "required", "all") \
-    .input(1, "dht", False, "required", "all") \
-    .input(2, "dct", False, "required", "all") \
-    .input(3, "it", False, "required", "all") \
-    .input(4, "jt", False, "required", "all") \
-    .input(5, "ft", False, "required", "all") \
-    .input(6, "ot", False, "required", "all") \
-    .input(7, "tanhct", False, "required", "all") \
-    .output(0, "dgate", False, "required", "all") \
-    .output(1, "dct_1", False, "required", "all") \
-    .dtype_format(DataType.F32_FracNZ, DataType.F32_FracNZ, DataType.F32_FracNZ, DataType.F32_FracNZ,
-                  DataType.F32_FracNZ, DataType.F32_FracNZ, DataType.F32_FracNZ, DataType.F32_FracNZ,
-                  DataType.F16_FracNZ, DataType.F32_FracNZ) \
-    .dtype_format(DataType.F16_FracNZ, DataType.F16_FracNZ, DataType.F16_FracNZ, DataType.F16_FracNZ,
-                  DataType.F16_FracNZ, DataType.F16_FracNZ, DataType.F16_FracNZ, DataType.F16_FracNZ,
-                  DataType.F16_FracNZ, DataType.F16_FracNZ) \
-    .get_op_info()
-
-
-@op_info_register(basic_lstm_cell_c_state_grad_op_info)
-def _basic_lstm_cell_c_state_grad_tbe():
-    """BasicLSTMCellCStateGrad TBE register"""
-    return