mindspore 2.2.14__cp37-cp37m-manylinux1_x86_64.whl → 2.3.0rc1__cp37-cp37m-manylinux1_x86_64.whl

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

mindspore/ops/_op_impl/tbe/cummin.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.
-# ============================================================================
-
-"""Cummin op"""
-from mindspore.ops.op_info_register import op_info_register, TBERegOp, DataType
-
-celu_op_info = TBERegOp("Cummin") \
-    .fusion_type("OPAQUE") \
-    .async_flag(False) \
-    .binfile_name("cummin.so") \
-    .compute_cost(10) \
-    .kernel_name("cummin") \
-    .partial_flag(True) \
-    .attr("axis", "required", "int", "all") \
-    .input(0, "x", False, "required", "all") \
-    .output(0, "y", False, "required", "all") \
-    .output(0, "argmin", False, "required", "all") \
-    .dtype_format(DataType.F32_Default, DataType.F32_Default, DataType.I32_Default) \
-    .dtype_format(DataType.F16_Default, DataType.F16_Default, DataType.I32_Default) \
-    .dtype_format(DataType.I32_Default, DataType.I32_Default, DataType.I32_Default) \
-    .dtype_format(DataType.I8_Default, DataType.I8_Default, DataType.I32_Default) \
-    .dtype_format(DataType.U8_Default, DataType.U8_Default, DataType.I32_Default) \
-    .get_op_info()
-
-
-@op_info_register(celu_op_info)
-def _cummin_tbe():
-    """Cummin TBE register"""
-    return

mindspore/ops/_op_impl/tbe/cumprod.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.
-# ============================================================================
-
-"""CumProd op"""
-from mindspore.ops.op_info_register import op_info_register, TBERegOp, DataType
-
-cumprop_op_info = TBERegOp("CumProd") \
-    .fusion_type("OPAQUE") \
-    .async_flag(False) \
-    .binfile_name("cumprod_d.so") \
-    .compute_cost(10) \
-    .kernel_name("cumprod_d") \
-    .partial_flag(True) \
-    .attr("axis", "optional", "int", "all", "0") \
-    .attr("exclusive", "optional", "bool", "all") \
-    .attr("reverse", "optional", "bool", "all") \
-    .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(cumprop_op_info)
-def _cumprop_tbe():
-    """CumProd TBE register"""
-    return

mindspore/ops/_op_impl/tbe/data_format_dim_map.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.
-# ============================================================================
-
-"""DataFormatDimMap op"""
-from mindspore.ops.op_info_register import op_info_register, TBERegOp, DataType
-
-data_format_dim_map_op_info = TBERegOp("DataFormatDimMap") \
-    .fusion_type("OPAQUE") \
-    .async_flag(False) \
-    .binfile_name("data_format_dim_map.so") \
-    .compute_cost(10) \
-    .kernel_name("data_format_dim_map") \
-    .partial_flag(True) \
-    .attr("dst_format", "optional", "str", "all") \
-    .attr("src_format", "optional", "str", "all") \
-    .input(0, "x", False, "required", "all") \
-    .output(0, "y", False, "required", "all") \
-    .dtype_format(DataType.I32_5HD, DataType.I32_5HD) \
-    .dtype_format(DataType.I32_Default, DataType.I32_Default) \
-    .get_op_info()
-
-
-@op_info_register(data_format_dim_map_op_info)
-def _data_format_dim_map_tbe():
-    """DataFormatDimMap TBE register"""
-    return

mindspore/ops/_op_impl/tbe/data_format_dim_map_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.
-# ============================================================================
-
-"""DataFormatDimMap op"""
-from mindspore.ops.op_info_register import op_info_register, TBERegOp, DataType
-
-data_format_dim_map_op_info = TBERegOp("DataFormatDimMap") \
-    .fusion_type("OPAQUE") \
-    .async_flag(False) \
-    .binfile_name("data_format_dim_map.so") \
-    .compute_cost(10) \
-    .kernel_name("data_format_dim_map") \
-    .partial_flag(True) \
-    .dynamic_shape(True) \
-    .dynamic_compile_static(True) \
-    .attr("dst_format", "optional", "str", "all") \
-    .attr("src_format", "optional", "str", "all") \
-    .input(0, "x", False, "required", "all") \
-    .output(0, "y", False, "required", "all") \
-    .dtype_format(DataType.I32_5HD, DataType.I32_5HD) \
-    .dtype_format(DataType.I32_Default, DataType.I32_Default) \
-    .get_op_info()
-
-
-@op_info_register(data_format_dim_map_op_info)
-def _data_format_dim_map_ds_tbe():
-    """DataFormatDimMap TBE register"""
-    return

mindspore/ops/_op_impl/tbe/deformable_offsets.py

@@ -1,45 +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.
-# ============================================================================
-
-"""DeformableOffsets op"""
-from mindspore.ops.op_info_register import op_info_register, TBERegOp, DataType
-
-deformable_offsets_op_info = TBERegOp("DeformableOffsets") \
-    .fusion_type("OPAQUE") \
-    .async_flag(False) \
-    .binfile_name("deformable_offsets.so") \
-    .compute_cost(10) \
-    .kernel_name("deformable_offsets") \
-    .partial_flag(True) \
-    .need_check_supported(True) \
-    .attr("strides", "required", "listInt", "all") \
-    .attr("pads", "required", "listInt", "all") \
-    .attr("ksize", "required", "listInt", "all") \
-    .attr("dilations", "optional", "listInt", "all", "1,1,1,1") \
-    .attr("data_format", "optional", "str", "all", "NHWC") \
-    .attr("deformable_groups", "optional", "int", "all", "1") \
-    .attr("modulated", "optional", "bool", "all", "true") \
-    .input(0, "x", False, "required", "all") \
-    .input(1, "offsets", False, "required", "all") \
-    .input(2, "helper", False, "optional", "all") \
-    .output(0, "y", False, "required", "all") \
-    .dtype_format(DataType.F32_NHWC, DataType.F32_NHWC, DataType.F32_NHWC, DataType.F32_NHWC) \
-    .get_op_info()
-
-
-@op_info_register(deformable_offsets_op_info)
-def _deformable_offsets_tbe():
-    """DeformableOffsets TBE register"""
-    return

mindspore/ops/_op_impl/tbe/deformable_offsets_grad.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.
-# ============================================================================
-
-"""DeformableOffsetsGrad op"""
-from mindspore.ops.op_info_register import op_info_register, TBERegOp, DataType
-
-deformable_offsets_grad_op_info = TBERegOp("DeformableOffsetsGrad") \
-    .fusion_type("OPAQUE") \
-    .async_flag(False) \
-    .binfile_name("deformable_offsets_grad.so") \
-    .compute_cost(10) \
-    .kernel_name("deformable_offsets_grad") \
-    .partial_flag(True) \
-    .need_check_supported(True) \
-    .attr("strides", "required", "listInt", "all") \
-    .attr("pads", "required", "listInt", "all") \
-    .attr("ksize", "required", "listInt", "all") \
-    .attr("dilations", "optional", "listInt", "all", "1,1,1,1") \
-    .attr("data_format", "optional", "str", "all", "NHWC") \
-    .attr("deformable_groups", "optional", "int", "all", "1") \
-    .attr("modulated", "optional", "bool", "all", "true") \
-    .input(0, "grad", False, "required", "all") \
-    .input(1, "x", False, "required", "all") \
-    .input(2, "offsets", False, "required", "all") \
-    .input(3, "helper", False, "optional", "all") \
-    .output(0, "grad_x", False, "required", "all") \
-    .output(1, "grad_offsets", False, "required", "all") \
-    .dtype_format(DataType.F32_NHWC, DataType.F32_NHWC, DataType.F32_NHWC, DataType.F32_NHWC,
-                  DataType.F32_NHWC, DataType.F32_NHWC) \
-    .get_op_info()
-
-
-@op_info_register(deformable_offsets_grad_op_info)
-def _deformable_offsets_grad_tbe():
-    """DeformableOffsetsGrad TBE register"""
-    return

mindspore/ops/_op_impl/tbe/depth_to_space_ds.py

@@ -1,49 +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.
-# ============================================================================
-
-"""DepthToSpace op"""
-from mindspore.ops.op_info_register import op_info_register, TBERegOp, DataType
-
-depth_to_space_ds_op_info = TBERegOp("DepthToSpace") \
-    .fusion_type("OPAQUE") \
-    .async_flag(False) \
-    .binfile_name("depth_to_space.so") \
-    .compute_cost(10) \
-    .kernel_name("depth_to_space") \
-    .partial_flag(True) \
-    .dynamic_compile_static(True) \
-    .dynamic_shape(True) \
-    .attr("block_size", "required", "int", "all") \
-    .attr("mode", "optional", "str", "all", "DCR") \
-    .attr("data_format", "optional", "str", "all", "NCHW") \
-    .input(0, "x", False, "required", "all") \
-    .output(0, "y", False, "required", "all") \
-    .dtype_format(DataType.F16_NCHW, DataType.F16_NCHW) \
-    .dtype_format(DataType.F32_NCHW, DataType.F32_NCHW) \
-    .dtype_format(DataType.I8_NCHW, DataType.I8_NCHW) \
-    .dtype_format(DataType.I16_NCHW, DataType.I16_NCHW) \
-    .dtype_format(DataType.I32_NCHW, DataType.I32_NCHW) \
-    .dtype_format(DataType.I64_NCHW, DataType.I64_NCHW) \
-    .dtype_format(DataType.U8_NCHW, DataType.U8_NCHW) \
-    .dtype_format(DataType.U16_NCHW, DataType.U16_NCHW) \
-    .dtype_format(DataType.U32_NCHW, DataType.U32_NCHW) \
-    .dtype_format(DataType.U64_NCHW, DataType.U64_NCHW) \
-    .get_op_info()
-
-
-@op_info_register(depth_to_space_ds_op_info)
-def _depth_to_space_ds_tbe():
-    """DepthToSpace TBE register"""
-    return

mindspore/ops/_op_impl/tbe/depthwise_conv2d.py

@@ -1,44 +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.
-# ============================================================================
-
-"""DepthwiseConv2D op"""
-from mindspore.ops.op_info_register import op_info_register, TBERegOp, DataType
-
-depthwise_conv2d_op_info = TBERegOp("DepthwiseConv2D") \
-    .fusion_type("CONVOLUTION") \
-    .async_flag(False) \
-    .binfile_name("depthwise_conv2d.so") \
-    .compute_cost(10) \
-    .kernel_name("depthwise_conv2d") \
-    .partial_flag(True) \
-    .attr("stride", "required", "listInt", "all", "[]") \
-    .attr("dilation", "required", "listInt", "all", "[]") \
-    .attr("pad_list", "required", "listInt", "all", "[]") \
-    .attr("format", "required", "str", "all") \
-    .attr("offset_a", "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_C1HWNCoC0, DataType.F16_Default, DataType.F16_Default,
-                  DataType.F16_5HD) \
-    .get_op_info()
-
-
-@op_info_register(depthwise_conv2d_op_info)
-def _depthwise_conv2d_tbe():
-    """DepthwiseConv2D TBE register"""
-    return

mindspore/ops/_op_impl/tbe/depthwise_conv2d_backprop_filter.py

@@ -1,41 +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.
-# ============================================================================
-
-"""DepthwiseConv2DBackpropFilter op"""
-from mindspore.ops.op_info_register import op_info_register, TBERegOp, DataType
-
-depthwise_conv2d_backprop_filter_op_info = TBERegOp("DepthwiseConv2dNativeBackpropFilter") \
-    .fusion_type("CONVOLUTION") \
-    .async_flag(False) \
-    .binfile_name("depthwise_conv2d_backprop_filter_d.so") \
-    .compute_cost(10) \
-    .kernel_name("depthwise_conv2d_backprop_filter_d") \
-    .partial_flag(True) \
-    .attr("filter_size", "required", "listInt", "all") \
-    .attr("stride", "required", "listInt", "all") \
-    .attr("dilation", "required", "listInt", "all") \
-    .attr("pad_list", "required", "listInt", "all") \
-    .attr("format", "required", "str", "all") \
-    .input(0, "input", False, "required", "all") \
-    .input(1, "out_backprop", False, "required", "all") \
-    .output(0, "filter_grad", False, "required", "all") \
-    .dtype_format(DataType.F16_5HD, DataType.F16_5HD, DataType.F32_C1HWNCoC0) \
-    .get_op_info()
-
-
-@op_info_register(depthwise_conv2d_backprop_filter_op_info)
-def _depthwise_conv2d_backprop_filter_tbe():
-    """DepthwiseConv2DBackpropFilter TBE register"""
-    return

mindspore/ops/_op_impl/tbe/depthwise_conv2d_backprop_input.py

@@ -1,41 +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.
-# ============================================================================
-
-"""DepthwiseConv2DBackpropInput op"""
-from mindspore.ops.op_info_register import op_info_register, TBERegOp, DataType
-
-depthwise_conv2d_backprop_input_op_info = TBERegOp("DepthwiseConv2dNativeBackpropInput") \
-    .fusion_type("CONVOLUTION") \
-    .async_flag(False) \
-    .binfile_name("depthwise_conv2d_backprop_input_d.so") \
-    .compute_cost(10) \
-    .kernel_name("depthwise_conv2d_backprop_input_d") \
-    .partial_flag(True) \
-    .attr("input_size", "required", "listInt", "all") \
-    .attr("stride", "required", "listInt", "all") \
-    .attr("dilation", "required", "listInt", "all") \
-    .attr("pad_list", "required", "listInt", "all") \
-    .attr("format", "required", "str", "all") \
-    .input(0, "filter", False, "required", "all") \
-    .input(1, "out_backprop", False, "required", "all") \
-    .output(0, "input_grad", False, "required", "all") \
-    .dtype_format(DataType.F16_C1HWNCoC0, DataType.F16_5HD, DataType.F16_5HD) \
-    .get_op_info()
-
-
-@op_info_register(depthwise_conv2d_backprop_input_op_info)
-def _depthwise_conv2d_backprop_input_tbe():
-    """DepthwiseConv2DBackpropInput TBE register"""
-    return

mindspore/ops/_op_impl/tbe/diag.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.
-# ============================================================================
-
-"""DiagD op"""
-from mindspore.ops.op_info_register import op_info_register, TBERegOp, DataType
-
-diag_d_op_info = TBERegOp("Diag") \
-    .fusion_type("ELEMWISE") \
-    .async_flag(False) \
-    .binfile_name("diag_d.so") \
-    .compute_cost(10) \
-    .kernel_name("diag_d") \
-    .partial_flag(True) \
-    .input(0, "x", False, "required", "all") \
-    .input(1, "assist", False, "required", "all") \
-    .output(0, "y", False, "required", "all") \
-    .dtype_format(DataType.F16_Default, DataType.F16_Default, DataType.F16_Default) \
-    .dtype_format(DataType.F32_Default, DataType.F32_Default, DataType.F32_Default) \
-    .dtype_format(DataType.I32_Default, DataType.I32_Default, DataType.I32_Default) \
-    .get_op_info()
-
-
-@op_info_register(diag_d_op_info)
-def _diag_tbe():
-    """DiagD TBE register"""
-    return

mindspore/ops/_op_impl/tbe/diag_part.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.
-# ============================================================================
-
-"""DiagPartD op"""
-from mindspore.ops.op_info_register import op_info_register, TBERegOp, DataType
-
-diag_part_d_op_info = TBERegOp("DiagPart") \
-    .fusion_type("ELEMWISE") \
-    .async_flag(False) \
-    .binfile_name("diag_part_d.so") \
-    .compute_cost(10) \
-    .kernel_name("diag_part_d") \
-    .partial_flag(True) \
-    .input(0, "x", False, "required", "all") \
-    .input(1, "assist", False, "required", "all") \
-    .output(0, "y", False, "required", "all") \
-    .dtype_format(DataType.F16_Default, DataType.F16_Default, DataType.F16_Default) \
-    .dtype_format(DataType.F32_Default, DataType.F32_Default, DataType.F32_Default) \
-    .dtype_format(DataType.I32_Default, DataType.I32_Default, DataType.I32_Default) \
-    .get_op_info()
-
-
-@op_info_register(diag_part_d_op_info)
-def _diag_part_tbe():
-    """DiagPartD TBE register"""
-    return

mindspore/ops/_op_impl/tbe/dilation.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.
-# ============================================================================
-
-"""Dilation op"""
-from mindspore.ops.op_info_register import op_info_register, TBERegOp, DataType
-
-dilation_op_info = TBERegOp("Dilation") \
-    .fusion_type("OPAQUE") \
-    .async_flag(False) \
-    .binfile_name("dilation.so") \
-    .compute_cost(10) \
-    .kernel_name("dilation") \
-    .partial_flag(True) \
-    .attr("dilations", "required", "listInt", "all") \
-    .attr("pads", "optional", "listInt", "all", "[]") \
-    .attr("padding_value", "optional", "float", "all", "0.0") \
-    .input(0, "x", False, "required", "all") \
-    .output(0, "y", False, "required", "all") \
-    .dtype_format(DataType.I8_5HD, DataType.I8_5HD) \
-    .dtype_format(DataType.F16_5HD, DataType.F16_5HD) \
-    .dtype_format(DataType.F32_5HD, DataType.F32_5HD) \
-    .get_op_info()
-
-
-@op_info_register(dilation_op_info)
-def _dilation_tbe():
-    """Dilation TBE register"""
-    return

mindspore/ops/_op_impl/tbe/div.py

@@ -1,41 +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.
-# ============================================================================
-
-"""Div op"""
-from mindspore.ops.op_info_register import op_info_register, TBERegOp, DataType
-
-div_op_info = TBERegOp("Div") \
-    .fusion_type("ELEMWISE") \
-    .async_flag(False) \
-    .binfile_name("truncate_div.so") \
-    .compute_cost(10) \
-    .kernel_name("truncate_div") \
-    .partial_flag(True) \
-    .input(0, "x1", False, "required", "all") \
-    .input(1, "x2", False, "required", "all") \
-    .output(0, "y", False, "required", "all") \
-    .op_pattern("broadcast") \
-    .dtype_format(DataType.I8_None, DataType.I8_None, DataType.I8_None) \
-    .dtype_format(DataType.U8_None, DataType.U8_None, DataType.U8_None) \
-    .dtype_format(DataType.I32_None, DataType.I32_None, DataType.I32_None) \
-    .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(div_op_info)
-def _div_tbe():
-    """Div TBE register"""
-    return