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

mindspore/ops/_op_impl/tbe/conv2d_backprop_input_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.
-# ============================================================================
-
-"""Conv2DBackpropInput op"""
-from mindspore.ops.op_info_register import op_info_register, TBERegOp, DataType
-
-conv2d_backprop_input_op_info = TBERegOp("Conv2DBackpropInput") \
-    .fusion_type("CONVOLUTION") \
-    .async_flag(False) \
-    .binfile_name("conv2d_backprop_input.so") \
-    .compute_cost(10) \
-    .kernel_name("conv2d_backprop_input") \
-    .partial_flag(True) \
-    .dynamic_shape(True) \
-    .attr("stride", "required", "listInt", "all") \
-    .attr("pad_list", "required", "listInt", "all") \
-    .attr("dilation", "required", "listInt", "all") \
-    .attr("groups", "optional", "int", "all", "1") \
-    .attr("format", "optional", "str", "all", "NHWC") \
-    .input(0, "out_backprop", False, "required", "all") \
-    .input(1, "filter", False, "required", "all") \
-    .input(2, "input_size", False, "required", "all") \
-    .output(0, "y", True, "required", "all") \
-    .dtype_format(DataType.F16_5HD, DataType.F16_FracZ, DataType.I32_Default, DataType.F16_5HD) \
-    .dtype_format(DataType.F16_5HD, DataType.F16_FracZ, DataType.I32_Default, DataType.F32_5HD) \
-    .get_op_info()
-
-
-@op_info_register(conv2d_backprop_input_op_info)
-def _conv2d_backprop_input_ds_tbe():
-    """Conv2DBackpropInput TBE register"""
-    return

mindspore/ops/_op_impl/tbe/conv2d_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.
-# ============================================================================
-
-"""Conv2D op"""
-from mindspore.ops.op_info_register import op_info_register, TBERegOp, DataType
-
-conv2d_op_info = TBERegOp("Conv2D") \
-    .fusion_type("CONVOLUTION") \
-    .async_flag(False) \
-    .binfile_name("conv2d.so") \
-    .compute_cost(10) \
-    .kernel_name("conv2d") \
-    .partial_flag(True) \
-    .dynamic_shape(True) \
-    .attr("stride", "required", "listInt", "all") \
-    .attr("pad_list", "required", "listInt", "all") \
-    .attr("dilation", "required", "listInt", "all") \
-    .attr("groups", "optional", "int", "all", "1") \
-    .attr("format", "optional", "str", "all", "NCHW") \
-    .attr("offset_x", "optional", "int", "all", "0") \
-    .input(0, "x", False, "required", "all") \
-    .input(1, "filter", False, "required", "all") \
-    .input(2, "bias", False, "optional", "all") \
-    .input(3, "offset_w", False, "optional", "all") \
-    .output(0, "y", True, "required", "all") \
-    .is_dynamic_format(True) \
-    .dtype_format(DataType.F16_None, DataType.F16_None, DataType.F16_None, DataType.I8_None, DataType.F16_None) \
-    .get_op_info()
-
-# .dtype_format(DataType.F16_5HD, DataType.F16_FracZ, DataType.F16_Default, DataType.I8_Default, DataType.F16_Default) ?
-
-@op_info_register(conv2d_op_info)
-def _conv2d_ds_tbe():
-    """Conv2D TBE register"""
-    return

mindspore/ops/_op_impl/tbe/conv2d_transpose.py

@@ -1,48 +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.
-# ============================================================================
-
-"""Conv2DTranspose op"""
-from mindspore.ops.op_info_register import op_info_register, TBERegOp, DataType
-
-conv2d_transpose_op_info = TBERegOp("Conv2DTranspose") \
-    .fusion_type("CONVOLUTION") \
-    .async_flag(False) \
-    .binfile_name("conv2d_transpose_d.so") \
-    .compute_cost(10) \
-    .kernel_name("conv2d_transpose_d") \
-    .partial_flag(True) \
-    .attr("input_sizes", "required", "listInt", "all") \
-    .attr("stride", "required", "listInt", "all") \
-    .attr("pad_list", "required", "listInt", "all") \
-    .attr("dilation", "optional", "listInt", "all", "1,1,1,1") \
-    .attr("groups", "optional", "int", "all", "1") \
-    .attr("format", "optional", "str", "all", "NHWC") \
-    .attr("output_padding", "optional", "listInt", "all", "0,0,0,0") \
-    .attr("offset_x", "optional", "int", "all", "0") \
-    .input(0, "x", False, "required", "all") \
-    .input(1, "filter", False, "required", "all") \
-    .input(2, "bias", False, "optional", "all") \
-    .input(3, "offset_w", False, "optional", "all") \
-    .output(0, "y", True, "required", "all") \
-    .dtype_format(DataType.F16_5HD, DataType.F16_FracZ, DataType.F16_Default, DataType.I8_Default, DataType.F16_5HD) \
-    .dtype_format(DataType.F16_5HD, DataType.F16_FracZ, DataType.F32_Default, DataType.I8_Default, DataType.F32_5HD) \
-    .dtype_format(DataType.I8_5HD, DataType.I8_FracZ, DataType.I32_Default, DataType.I8_Default, DataType.I32_5HD) \
-    .get_op_info()
-
-
-@op_info_register(conv2d_transpose_op_info)
-def _conv2d_transpose_tbe():
-    """Conv2DTranspose TBE register"""
-    return

mindspore/ops/_op_impl/tbe/conv3d.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.
-# ============================================================================
-
-"""Conv3D op"""
-from mindspore.ops.op_info_register import op_info_register, TBERegOp, DataType
-
-conv3d_op_info = TBERegOp("Conv3D") \
-    .fusion_type("CONVOLUTION") \
-    .async_flag(False) \
-    .binfile_name("conv3d.so") \
-    .compute_cost(10) \
-    .kernel_name("conv3d") \
-    .partial_flag(True) \
-    .attr("strides", "required", "listInt", "all") \
-    .attr("pad_list", "required", "listInt", "all") \
-    .attr("dilations", "optional", "listInt", "all", "1,1,1,1,1") \
-    .attr("groups", "optional", "int", "all", "1") \
-    .attr("format", "optional", "str", "all", "NDHWC") \
-    .attr("offset_x", "optional", "int", "all", "0") \
-    .input(0, "x", False, "required", "all") \
-    .input(1, "filter", False, "required", "all") \
-    .input(2, "bias", False, "optional", "all") \
-    .input(3, "offset_w", False, "optional", "all") \
-    .output(0, "y", True, "required", "all") \
-    .dtype_format(DataType.F16_NDC1HWC0, DataType.F16_FRACTAL_Z_3D, DataType.F16_Default,
-                  DataType.I8_Default, DataType.F16_NDC1HWC0) \
-    .get_op_info()
-
-
-@op_info_register(conv3d_op_info)
-def _conv3d_tbe():
-    """Conv3D TBE register"""
-    return

mindspore/ops/_op_impl/tbe/conv3d_backprop_filter.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.
-# ============================================================================
-
-"""Conv3DBackpropFilter op"""
-from mindspore.ops.op_info_register import op_info_register, TBERegOp, DataType
-
-conv3d_backprop_filter_op_info = TBERegOp("Conv3DBackpropFilter") \
-    .fusion_type("CONVOLUTION") \
-    .async_flag(False) \
-    .binfile_name("conv3d_backprop_filter_d.so") \
-    .compute_cost(10) \
-    .kernel_name("conv3d_backprop_filter_d") \
-    .partial_flag(True) \
-    .attr("filter_size", "required", "listInt", "all") \
-    .attr("strides", "required", "listInt", "all") \
-    .attr("pad_list", "required", "listInt", "all") \
-    .attr("dilations", "required", "listInt", "all") \
-    .attr("groups", "optional", "int", "all") \
-    .attr("format", "optional", "str", "all") \
-    .input(0, "x", False, "required", "all") \
-    .input(1, "out_backprop", False, "required", "all") \
-    .output(0, "y", True, "required", "all") \
-    .dtype_format(DataType.F16_NDC1HWC0, DataType.F16_NDC1HWC0, DataType.F32_FRACTAL_Z_3D) \
-    .get_op_info()
-
-
-@op_info_register(conv3d_backprop_filter_op_info)
-def _conv3d_backprop_filter_tbe():
-    """Conv3DBackpropFilter TBE register"""
-    return

mindspore/ops/_op_impl/tbe/conv3d_backprop_input.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.
-# ============================================================================
-
-"""Conv3DBackpropInput op"""
-from mindspore.ops.op_info_register import op_info_register, TBERegOp, DataType
-
-conv3d_backprop_input_op_info = TBERegOp("Conv3DBackpropInput") \
-    .fusion_type("CONVOLUTION") \
-    .async_flag(False) \
-    .binfile_name("conv3d_backprop_input_d.so") \
-    .compute_cost(10) \
-    .kernel_name("conv3d_backprop_input_d") \
-    .partial_flag(True) \
-    .attr("input_size", "required", "listInt", "all") \
-    .attr("strides", "required", "listInt", "all") \
-    .attr("pad_list", "required", "listInt", "all") \
-    .attr("dilations", "required", "listInt", "all") \
-    .attr("groups", "optional", "int", "all") \
-    .attr("format", "optional", "str", "all") \
-    .input(0, "filter", False, "required", "all") \
-    .input(1, "out_backprop", False, "required", "all") \
-    .output(0, "y", True, "required", "all") \
-    .dtype_format(DataType.F16_FRACTAL_Z_3D, DataType.F16_NDC1HWC0, DataType.F16_NDC1HWC0) \
-    .get_op_info()
-
-
-@op_info_register(conv3d_backprop_input_op_info)
-def _conv3d_backprop_input_tbe():
-    """Conv3DBackpropInput TBE register"""
-    return

mindspore/ops/_op_impl/tbe/conv3d_transpose.py

@@ -1,47 +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.
-# ============================================================================
-
-"""Conv3DTranspose op"""
-from mindspore.ops.op_info_register import op_info_register, TBERegOp, DataType
-
-conv3d_transpose_op_info = TBERegOp("Conv3DTranspose") \
-    .fusion_type("CONVOLUTION") \
-    .async_flag(False) \
-    .binfile_name("conv3d_transpose_d.so") \
-    .compute_cost(10) \
-    .kernel_name("conv3d_transpose_d") \
-    .partial_flag(True) \
-    .attr("input_size", "required", "listInt", "all") \
-    .attr("strides", "required", "listInt", "all") \
-    .attr("pad_list", "required", "listInt", "all") \
-    .attr("dilations", "optional", "listInt", "all") \
-    .attr("groups", "optional", "int", "all") \
-    .attr("format", "optional", "str", "all") \
-    .attr("output_padding", "optional", "listInt", "all") \
-    .attr("offset_x", "optional", "int", "all", "0") \
-    .input(0, "x", False, "required", "all") \
-    .input(1, "filter", False, "required", "all") \
-    .input(2, "bias", False, "optional", "all") \
-    .input(3, "offset_w", False, "optional", "all") \
-    .output(0, "y", True, "required", "all") \
-    .dtype_format(DataType.F16_NDC1HWC0, DataType.F16_FRACTAL_Z_3D, DataType.F16_Default, DataType.I8_Default,
-                  DataType.F16_NDC1HWC0) \
-    .get_op_info()
-
-
-@op_info_register(conv3d_transpose_op_info)
-def _conv3d_transpose_tbe():
-    """Conv3DTranspose TBE register"""
-    return

mindspore/ops/_op_impl/tbe/conv3d_transpose_ds.py

@@ -1,48 +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.
-# ============================================================================
-
-"""Conv3DTranspose op"""
-from mindspore.ops.op_info_register import op_info_register, TBERegOp, DataType
-
-conv3d_transpose_ds_op_info = TBERegOp("Conv3DTranspose") \
-    .fusion_type("CONVOLUTION") \
-    .async_flag(False) \
-    .binfile_name("conv3d_transpose_d.so") \
-    .compute_cost(10) \
-    .kernel_name("conv3d_transpose_d") \
-    .partial_flag(True) \
-    .dynamic_shape(True) \
-    .attr("input_size", "required", "listInt", "all") \
-    .attr("strides", "required", "listInt", "all") \
-    .attr("pad_list", "required", "listInt", "all") \
-    .attr("dilations", "optional", "listInt", "all") \
-    .attr("groups", "optional", "int", "all") \
-    .attr("format", "optional", "str", "all") \
-    .attr("output_padding", "optional", "listInt", "all") \
-    .attr("offset_x", "optional", "int", "all", "0") \
-    .input(0, "x", False, "required", "all") \
-    .input(1, "filter", False, "required", "all") \
-    .input(2, "bias", False, "optional", "all") \
-    .input(3, "offset_w", False, "optional", "all") \
-    .output(0, "y", True, "required", "all") \
-    .dtype_format(DataType.F16_NDC1HWC0, DataType.F16_FRACTAL_Z_3D, DataType.F16_Default, DataType.I8_Default,
-                  DataType.F16_NDC1HWC0) \
-    .get_op_info()
-
-
-@op_info_register(conv3d_transpose_ds_op_info)
-def _conv3d_transpose_ds_tbe():
-    """Conv3DTranspose TBE register"""
-    return

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

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

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

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

mindspore/ops/_op_impl/tbe/ctc_loss_v2.py

@@ -1,42 +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.
-# ============================================================================
-
-"""CTC_LossV2 op"""
-from mindspore.ops.op_info_register import op_info_register, TBERegOp, DataType
-
-ctc_loss_v2_info = TBERegOp("CTCLossV2") \
-    .fusion_type("OPAQUE") \
-    .async_flag(False) \
-    .binfile_name("ctc_loss_v2.so") \
-    .compute_cost(10) \
-    .kernel_name("ctc_loss_v2") \
-    .partial_flag(True) \
-    .attr("blank", "optional", "int", "all", "0") \
-    .attr("reduction", "optional", "str", "all", "none") \
-    .attr("zero_infinity", "optional", "bool", "all", "false") \
-    .input(0, "log_probs", False, "required", "all") \
-    .input(1, "targets", False, "required", "all") \
-    .input(2, "input_lengths", False, "required", "all") \
-    .input(3, "target_lengths", False, "required", "all") \
-    .output(0, "neg_log_likelihood", False, "required", "all") \
-    .output(1, "log_alpha", False, "required", "all") \
-    .dtype_format(DataType.F32_Default, DataType.I32_Default, DataType.I32_Default, DataType.I32_Default,
-                  DataType.F32_Default, DataType.F32_Default) \
-    .get_op_info()
-
-@op_info_register(ctc_loss_v2_info)
-def _ctc_loss_v2_tbe():
-    """CTCLossV2 TBE register"""
-    return

mindspore/ops/_op_impl/tbe/ctc_loss_v2_grad.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.
-# ============================================================================
-
-"""CTC_LossV2Grad op"""
-from mindspore.ops.op_info_register import op_info_register, TBERegOp, DataType
-
-ctc_loss_v2_grad_info = TBERegOp("CTCLossV2Grad") \
-    .fusion_type("OPAQUE") \
-    .async_flag(False) \
-    .binfile_name("ctc_loss_v2_grad.so") \
-    .compute_cost(10) \
-    .kernel_name("ctc_loss_v2_grad") \
-    .partial_flag(True) \
-    .attr("blank", "optional", "int", "all", "0") \
-    .attr("reduction", "optional", "str", "all", "none") \
-    .attr("zero_infinity", "optional", "bool", "all", "false") \
-    .input(0, "grad_out", False, "required", "all") \
-    .input(1, "log_probs", False, "required", "all") \
-    .input(2, "targets", False, "required", "all") \
-    .input(3, "input_lengths", False, "required", "all") \
-    .input(4, "target_lengths", False, "required", "all") \
-    .input(5, "neg_log_likelihood", False, "required", "all") \
-    .input(6, "log_alpha", False, "required", "all") \
-    .output(0, "grad", False, "required", "all") \
-    .dtype_format(DataType.F32_Default, DataType.F32_Default, DataType.I32_Default, DataType.I32_Default,
-                  DataType.I32_Default, DataType.F32_Default, DataType.F32_Default, DataType.F32_Default) \
-    .get_op_info()
-
-@op_info_register(ctc_loss_v2_grad_info)
-def _ctc_loss_v2_grad_tbe():
-    """CTCLossV2Grad TBE register"""
-    return

mindspore/ops/_op_impl/tbe/cum_sum.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.
-# ============================================================================
-
-"""CumSum op"""
-from mindspore.ops.op_info_register import op_info_register, TBERegOp, DataType
-
-cum_sum_op_info = TBERegOp("CumSum") \
-    .fusion_type("OPAQUE") \
-    .async_flag(False) \
-    .binfile_name("cumsum_d.so") \
-    .compute_cost(10) \
-    .kernel_name("cumsum_d") \
-    .partial_flag(True) \
-    .attr("axis", "optional", "int", "all", "0") \
-    .attr("exclusive", "optional", "bool", "true,false", "false") \
-    .attr("reverse", "optional", "bool", "true,false", "false") \
-    .input(0, "x", False, "required", "all") \
-    .output(0, "y", False, "required", "all") \
-    .dtype_format(DataType.I32_Default, DataType.I32_Default) \
-    .dtype_format(DataType.F32_Default, DataType.F32_Default) \
-    .dtype_format(DataType.F16_Default, DataType.F16_Default) \
-    .dtype_format(DataType.I8_Default, DataType.I8_Default) \
-    .dtype_format(DataType.U8_Default, DataType.U8_Default) \
-    .get_op_info()
-
-
-@op_info_register(cum_sum_op_info)
-def _cum_sum_tbe():
-    """CumSum TBE register"""
-    return