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

mindspore/ops/_op_impl/tbe/npu_alloc_float_status.py

@@ -1,34 +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.
-# ============================================================================
-
-"""NPUAllocFloatStatus op"""
-from mindspore.ops.op_info_register import op_info_register, TBERegOp, DataType
-
-npu_alloc_float_status_op_info = TBERegOp("NPUAllocFloatStatus") \
-    .fusion_type("OPAQUE") \
-    .async_flag(False) \
-    .binfile_name("n_p_u_alloc_float_status.so") \
-    .compute_cost(10) \
-    .kernel_name("n_p_u_alloc_float_status") \
-    .partial_flag(True) \
-    .output(0, "data", False, "required", "all") \
-    .dtype_format(DataType.F32_Default) \
-    .get_op_info()
-
-
-@op_info_register(npu_alloc_float_status_op_info)
-def _npu_alloc_float_status_tbe():
-    """NPUAllocFloatStatus TBE register"""
-    return

mindspore/ops/_op_impl/tbe/npu_clear_float_status.py

@@ -1,35 +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.
-# ============================================================================
-
-"""NPUClearFloatStatus op"""
-from mindspore.ops.op_info_register import op_info_register, TBERegOp, DataType
-
-npu_clear_float_status_op_info = TBERegOp("NPUClearFloatStatus") \
-    .fusion_type("OPAQUE") \
-    .async_flag(False) \
-    .binfile_name("n_p_u_clear_float_status.so") \
-    .compute_cost(10) \
-    .kernel_name("n_p_u_clear_float_status") \
-    .partial_flag(True) \
-    .input(0, "addr", False, "required", "all") \
-    .output(0, "data", False, "required", "all") \
-    .dtype_format(DataType.F32_Default, DataType.F32_Default) \
-    .get_op_info()
-
-
-@op_info_register(npu_clear_float_status_op_info)
-def _npu_clear_float_status_tbe():
-    """NPUClearFloatStatus TBE register"""
-    return

mindspore/ops/_op_impl/tbe/npu_clear_float_status_v2.py

@@ -1,35 +0,0 @@
-# Copyright 2023 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.
-# ============================================================================
-
-"""NPUClearFloatStatusV2 op"""
-from mindspore.ops.op_info_register import op_info_register, TBERegOp, DataType
-
-npu_clear_float_status_v2_op_info = TBERegOp("NPUClearFloatStatusV2") \
-    .fusion_type("OPAQUE") \
-    .async_flag(False) \
-    .binfile_name("n_p_u_clear_float_status_v2.so") \
-    .compute_cost(10) \
-    .kernel_name("n_p_u_clear_float_status_v2") \
-    .partial_flag(True) \
-    .input(0, "addr", False, "required", "all") \
-    .output(0, "data", False, "required", "all") \
-    .dtype_format(DataType.I32_Default, DataType.I32_Default) \
-    .get_op_info()
-
-
-@op_info_register(npu_clear_float_status_v2_op_info)
-def _npu_clear_float_status_v2_tbe():
-    """NPUClearFloatStatusV2 TBE register"""
-    return

mindspore/ops/_op_impl/tbe/npu_get_float_status.py

@@ -1,35 +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.
-# ============================================================================
-
-"""NPUGetFloatStatus op"""
-from mindspore.ops.op_info_register import op_info_register, TBERegOp, DataType
-
-npu_get_float_status_op_info = TBERegOp("NPUGetFloatStatus") \
-    .fusion_type("ELEMWISE") \
-    .async_flag(False) \
-    .binfile_name("n_p_u_get_float_status.so") \
-    .compute_cost(10) \
-    .kernel_name("n_p_u_get_float_status") \
-    .partial_flag(True) \
-    .input(0, "addr", False, "required", "all") \
-    .output(0, "data", False, "required", "all") \
-    .dtype_format(DataType.F32_Default, DataType.F32_Default) \
-    .get_op_info()
-
-
-@op_info_register(npu_get_float_status_op_info)
-def _npu_get_float_status_tbe():
-    """NPUGetFloatStatus TBE register"""
-    return

mindspore/ops/_op_impl/tbe/npu_get_float_status_v2.py

@@ -1,35 +0,0 @@
-# Copyright 2023 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.
-# ============================================================================
-
-"""NPUGetFloatStatusV2 op"""
-from mindspore.ops.op_info_register import op_info_register, TBERegOp, DataType
-
-npu_get_float_status_v2_op_info = TBERegOp("NPUGetFloatStatusV2") \
-    .fusion_type("ELEMWISE") \
-    .async_flag(False) \
-    .binfile_name("n_p_u_get_float_status_v2.so") \
-    .compute_cost(10) \
-    .kernel_name("n_p_u_get_float_status_v2") \
-    .partial_flag(True) \
-    .input(0, "addr", False, "required", "all") \
-    .output(0, "data", False, "required", "all") \
-    .dtype_format(DataType.I32_Default, DataType.I32_Default) \
-    .get_op_info()
-
-
-@op_info_register(npu_get_float_status_v2_op_info)
-def _npu_get_float_status_v2_tbe():
-    """NPUGetFloatStatusV2 TBE register"""
-    return

mindspore/ops/_op_impl/tbe/one_hot.py

@@ -1,48 +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.
-# ============================================================================
-
-"""OneHot op"""
-from mindspore.ops.op_info_register import op_info_register, TBERegOp, DataType
-
-one_hot_op_info = TBERegOp("OneHot") \
-    .fusion_type("OPAQUE") \
-    .async_flag(False) \
-    .binfile_name("one_hot.so") \
-    .compute_cost(10) \
-    .kernel_name("one_hot_d") \
-    .partial_flag(True) \
-    .attr("depth", "required", "int", "all") \
-    .attr("axis", "required", "int", "all", "-1") \
-    .input(0, "x", False, "required", "all") \
-    .input(1, "on_value", False, "required", "all") \
-    .input(2, "off_value", False, "required", "all") \
-    .output(0, "y", False, "required", "all") \
-    .dtype_format(DataType.U8_Default, DataType.I8_Default, DataType.I8_Default, DataType.I8_Default) \
-    .dtype_format(DataType.U8_Default, DataType.U8_Default, DataType.U8_Default, DataType.U8_Default) \
-    .dtype_format(DataType.U8_Default, DataType.I32_Default, DataType.I32_Default, DataType.I32_Default) \
-    .dtype_format(DataType.U8_Default, DataType.F16_Default, DataType.F16_Default, DataType.F16_Default) \
-    .dtype_format(DataType.U8_Default, DataType.F32_Default, DataType.F32_Default, DataType.F32_Default) \
-    .dtype_format(DataType.I32_Default, DataType.I8_Default, DataType.I8_Default, DataType.I8_Default) \
-    .dtype_format(DataType.I32_Default, DataType.U8_Default, DataType.U8_Default, DataType.U8_Default) \
-    .dtype_format(DataType.I32_Default, DataType.I32_Default, DataType.I32_Default, DataType.I32_Default) \
-    .dtype_format(DataType.I32_Default, DataType.F16_Default, DataType.F16_Default, DataType.F16_Default) \
-    .dtype_format(DataType.I32_Default, DataType.F32_Default, DataType.F32_Default, DataType.F32_Default) \
-    .get_op_info()
-
-
-@op_info_register(one_hot_op_info)
-def _one_hot_tbe():
-    """OneHot TBE register"""
-    return

mindspore/ops/_op_impl/tbe/one_hot_ds.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.
-# ============================================================================
-
-"""OneHot op"""
-from mindspore.ops.op_info_register import op_info_register, TBERegOp, DataType
-
-one_hot_op_info = TBERegOp("OneHot") \
-    .fusion_type("OPAQUE") \
-    .async_flag(False) \
-    .binfile_name("one_hot.so") \
-    .compute_cost(10) \
-    .kernel_name("one_hot") \
-    .partial_flag(True) \
-    .dynamic_shape(True) \
-    .attr("axis", "required", "int", "all", "-1") \
-    .input(0, "x", False, "required", "all") \
-    .input(1, "depth", False, "required", "all") \
-    .input(2, "on_value", False, "required", "all") \
-    .input(3, "off_value", False, "required", "all") \
-    .output(0, "y", False, "required", "all") \
-    .dtype_format(DataType.I32_Default, DataType.I32_Default, DataType.I32_Default, DataType.I32_Default,
-                  DataType.I32_Default) \
-    .dtype_format(DataType.I32_Default, DataType.I32_Default, DataType.F16_Default, DataType.F16_Default,
-                  DataType.F16_Default) \
-    .dtype_format(DataType.I32_Default, DataType.I32_Default, DataType.F32_Default, DataType.F32_Default,
-                  DataType.F32_Default) \
-    .get_op_info()
-
-
-@op_info_register(one_hot_op_info)
-def _one_hot_ds_tbe():
-    """OneHot TBE register"""
-    return

mindspore/ops/_op_impl/tbe/ones_like.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.
-# ============================================================================
-
-"""OnesLike op"""
-from mindspore.ops.op_info_register import op_info_register, TBERegOp, DataType
-
-ones_like_op_info = TBERegOp("OnesLike") \
-    .fusion_type("OPAQUE") \
-    .async_flag(False) \
-    .binfile_name("ones_like.so") \
-    .compute_cost(10) \
-    .kernel_name("ones_like") \
-    .partial_flag(True) \
-    .input(0, "x", False, "required", "all") \
-    .output(0, "y", False, "required", "all") \
-    .op_pattern("formatAgnostic") \
-    .dtype_format(DataType.U8_None, DataType.U8_None) \
-    .dtype_format(DataType.I8_None, DataType.I8_None) \
-    .dtype_format(DataType.I32_None, DataType.I32_None) \
-    .dtype_format(DataType.F16_None, DataType.F16_None) \
-    .dtype_format(DataType.F32_None, DataType.F32_None) \
-    .get_op_info()
-
-
-@op_info_register(ones_like_op_info)
-def _ones_like_tbe():
-    """OnesLike TBE register"""
-    return

mindspore/ops/_op_impl/tbe/ones_like_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.
-# ============================================================================
-
-"""OnesLike op"""
-from mindspore.ops.op_info_register import op_info_register, TBERegOp, DataType
-
-ones_like_op_info = TBERegOp("OnesLike") \
-    .fusion_type("OPAQUE") \
-    .async_flag(False) \
-    .binfile_name("ones_like.so") \
-    .compute_cost(10) \
-    .kernel_name("ones_like") \
-    .partial_flag(True) \
-    .dynamic_shape(True) \
-    .input(0, "x", False, "required", "all") \
-    .output(0, "y", False, "required", "all") \
-    .op_pattern("formatAgnostic") \
-    .dtype_format(DataType.U8_None, DataType.U8_None) \
-    .dtype_format(DataType.I8_None, DataType.I8_None) \
-    .dtype_format(DataType.I32_None, DataType.I32_None) \
-    .dtype_format(DataType.F16_None, DataType.F16_None) \
-    .dtype_format(DataType.F32_None, DataType.F32_None) \
-    .get_op_info()
-
-
-@op_info_register(ones_like_op_info)
-def _ones_like_ds_tbe():
-    """OnesLike TBE register"""
-    return

mindspore/ops/_op_impl/tbe/p_s_r_o_i_pooling.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.
-# ============================================================================
-
-"""PSROIPooling op"""
-from mindspore.ops.op_info_register import op_info_register, TBERegOp, DataType
-
-p_s_r_o_i_pooling_op_info = TBERegOp("PSROIPooling") \
-    .fusion_type("OPAQUE") \
-    .async_flag(False) \
-    .binfile_name("p_s_r_o_i_pooling_v2.so") \
-    .compute_cost(10) \
-    .kernel_name("p_s_r_o_i_pooling_v2") \
-    .partial_flag(True) \
-    .attr("output_dim", "required", "int", "all") \
-    .attr("group_size", "required", "int", "all") \
-    .attr("spatial_scale", "required", "float", "all") \
-    .input(0, "x", False, "required", "all") \
-    .input(1, "rois", False, "required", "all") \
-    .output(0, "y", False, "required", "all") \
-    .dtype_format(DataType.F16_5HD, DataType.F16_Default, DataType.F16_5HD) \
-    .dtype_format(DataType.F32_5HD, DataType.F32_Default, DataType.F32_5HD) \
-    .get_op_info()
-
-
-@op_info_register(p_s_r_o_i_pooling_op_info)
-def _p_s_r_o_i_pooling_tbe():
-    """PSROIPooling TBE register"""
-    return

mindspore/ops/_op_impl/tbe/p_s_r_o_i_pooling_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.
-# ============================================================================
-
-"""PSROIPoolingGrad op"""
-from mindspore.ops.op_info_register import op_info_register, TBERegOp, DataType
-
-p_s_r_o_i_pooling_grad_op_info = TBERegOp("PSROIPoolingGrad") \
-    .fusion_type("OPAQUE") \
-    .async_flag(False) \
-    .binfile_name("p_s_r_o_i_pooling_grad_v2_d.so") \
-    .compute_cost(10) \
-    .kernel_name("p_s_r_o_i_pooling_grad_v2_d") \
-    .partial_flag(True) \
-    .attr("output_dim", "required", "int", "all") \
-    .attr("group_size", "required", "int", "all") \
-    .attr("spatial_scale", "required", "float", "all") \
-    .attr("input_size", "required", "listInt", "all") \
-    .input(0, "x", False, "required", "all") \
-    .input(1, "rois", False, "required", "all") \
-    .output(0, "y", False, "required", "all") \
-    .dtype_format(DataType.F32_5HD, DataType.F32_Default, DataType.F32_5HD) \
-    .get_op_info()
-
-
-@op_info_register(p_s_r_o_i_pooling_grad_op_info)
-def _p_s_r_o_i_pooling_grad_tbe():
-    """PSROIPoolingGrad TBE register"""
-    return

mindspore/ops/_op_impl/tbe/pack.py

@@ -1,58 +0,0 @@
-# Copyright 2020 Huawei Technologies Co., Ltd
-#
-# Licensed under the Apache License, Version 2.0 (the "License");
-# you may not use this file except in compliance with the License.
-# You may obtain a copy of the License at
-#
-# http://www.apache.org/licenses/LICENSE-2.0
-#
-# Unless required by applicable law or agreed to in writing, software
-# distributed under the License is distributed on an "AS IS" BASIS,
-# WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
-# See the License for the specific language governing permissions and
-# limitations under the License.
-# ============================================================================
-
-"""Pack op"""
-from mindspore.ops.op_info_register import op_info_register, TBERegOp, DataType
-
-stack_op_info = TBERegOp("Stack") \
-    .fusion_type("OPAQUE") \
-    .async_flag(False) \
-    .binfile_name("pack.so") \
-    .compute_cost(10) \
-    .kernel_name("pack") \
-    .partial_flag(True) \
-    .need_check_supported(False) \
-    .attr("axis", "optional", "int", "all") \
-    .input(0, "x", False, "dynamic", "all") \
-    .output(0, "y", False, "required", "all") \
-    .dtype_format(DataType.I8_Default, DataType.I8_Default) \
-    .dtype_format(DataType.I16_Default, DataType.I16_Default) \
-    .dtype_format(DataType.I32_Default, DataType.I32_Default) \
-    .dtype_format(DataType.I64_Default, DataType.I64_Default) \
-    .dtype_format(DataType.U8_Default, DataType.U8_Default) \
-    .dtype_format(DataType.U16_Default, DataType.U16_Default) \
-    .dtype_format(DataType.U32_Default, DataType.U32_Default) \
-    .dtype_format(DataType.U64_Default, DataType.U64_Default) \
-    .dtype_format(DataType.F16_Default, DataType.F16_Default) \
-    .dtype_format(DataType.F32_Default, DataType.F32_Default) \
-    .dtype_format(DataType.BOOL_Default, DataType.BOOL_Default) \
-    .dtype_format(DataType.I8_NDHWC, DataType.I8_NDHWC) \
-    .dtype_format(DataType.I16_NDHWC, DataType.I16_NDHWC) \
-    .dtype_format(DataType.I32_NDHWC, DataType.I32_NDHWC) \
-    .dtype_format(DataType.I64_NDHWC, DataType.I64_NDHWC) \
-    .dtype_format(DataType.U8_NDHWC, DataType.U8_NDHWC) \
-    .dtype_format(DataType.U16_NDHWC, DataType.U16_NDHWC) \
-    .dtype_format(DataType.U32_NDHWC, DataType.U32_NDHWC) \
-    .dtype_format(DataType.U64_NDHWC, DataType.U64_NDHWC) \
-    .dtype_format(DataType.F16_NDHWC, DataType.F16_NDHWC) \
-    .dtype_format(DataType.F32_NDHWC, DataType.F32_NDHWC) \
-    .dtype_format(DataType.BOOL_NDHWC, DataType.BOOL_NDHWC) \
-    .get_op_info()
-
-
-@op_info_register(stack_op_info)
-def _pack_tbe():
-    """Pack TBE register"""
-    return

mindspore/ops/_op_impl/tbe/pack_ds.py

@@ -1,59 +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.
-# ============================================================================
-
-"""Pack op"""
-from mindspore.ops.op_info_register import op_info_register, TBERegOp, DataType
-
-stack_op_info = TBERegOp("Stack") \
-    .fusion_type("OPAQUE") \
-    .async_flag(False) \
-    .binfile_name("pack.so") \
-    .compute_cost(10) \
-    .kernel_name("pack") \
-    .partial_flag(True) \
-    .need_check_supported(False) \
-    .attr("axis", "optional", "int", "all") \
-    .input(0, "x", False, "dynamic", "all") \
-    .output(0, "y", False, "required", "all") \
-    .dynamic_shape(True) \
-    .dtype_format(DataType.I8_Default, DataType.I8_Default) \
-    .dtype_format(DataType.I16_Default, DataType.I16_Default) \
-    .dtype_format(DataType.I32_Default, DataType.I32_Default) \
-    .dtype_format(DataType.I64_Default, DataType.I64_Default) \
-    .dtype_format(DataType.U8_Default, DataType.U8_Default) \
-    .dtype_format(DataType.U16_Default, DataType.U16_Default) \
-    .dtype_format(DataType.U32_Default, DataType.U32_Default) \
-    .dtype_format(DataType.U64_Default, DataType.U64_Default) \
-    .dtype_format(DataType.F16_Default, DataType.F16_Default) \
-    .dtype_format(DataType.F32_Default, DataType.F32_Default) \
-    .dtype_format(DataType.BOOL_Default, DataType.BOOL_Default) \
-    .dtype_format(DataType.I8_NDHWC, DataType.I8_NDHWC) \
-    .dtype_format(DataType.I16_NDHWC, DataType.I16_NDHWC) \
-    .dtype_format(DataType.I32_NDHWC, DataType.I32_NDHWC) \
-    .dtype_format(DataType.I64_NDHWC, DataType.I64_NDHWC) \
-    .dtype_format(DataType.U8_NDHWC, DataType.U8_NDHWC) \
-    .dtype_format(DataType.U16_NDHWC, DataType.U16_NDHWC) \
-    .dtype_format(DataType.U32_NDHWC, DataType.U32_NDHWC) \
-    .dtype_format(DataType.U64_NDHWC, DataType.U64_NDHWC) \
-    .dtype_format(DataType.F16_NDHWC, DataType.F16_NDHWC) \
-    .dtype_format(DataType.F32_NDHWC, DataType.F32_NDHWC) \
-    .dtype_format(DataType.BOOL_NDHWC, DataType.BOOL_NDHWC) \
-    .get_op_info()
-
-
-@op_info_register(stack_op_info)
-def _pack_ds_tbe():
-    """Pack TBE register"""
-    return

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

mindspore/ops/_op_impl/tbe/pad_d_ds.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.
-# ============================================================================
-
-"""Pad op"""
-from mindspore.ops.op_info_register import op_info_register, TBERegOp, DataType
-
-pad_d_op_info = TBERegOp("Pad") \
-    .fusion_type("OPAQUE") \
-    .async_flag(False) \
-    .binfile_name("pad_d.so") \
-    .compute_cost(10) \
-    .kernel_name("pad_d") \
-    .partial_flag(True) \
-    .attr("paddings", "optional", "listListInt", "all") \
-    .dynamic_shape(True) \
-    .input(0, "x", False, "required", "all") \
-    .output(0, "y", False, "required", "all") \
-    .dtype_format(DataType.I8_Default, DataType.I8_Default) \
-    .dtype_format(DataType.U8_Default, DataType.U8_Default) \
-    .dtype_format(DataType.I32_Default, DataType.I32_Default) \
-    .dtype_format(DataType.F16_Default, DataType.F16_Default) \
-    .dtype_format(DataType.F32_Default, DataType.F32_Default) \
-    .get_op_info()
-
-
-@op_info_register(pad_d_op_info)
-def _pad_d_ds_tbe():
-    """Pad TBE register"""
-    return