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

mindspore/ops/_op_impl/tbe/matmul.py

@@ -1,53 +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.
-# ============================================================================
-
-"""MatMul op"""
-from mindspore.ops.op_info_register import op_info_register, TBERegOp, DataType
-
-matmul_op_info = TBERegOp("MatMul") \
-    .fusion_type("OPAQUE") \
-    .async_flag(False) \
-    .binfile_name("mat_mul.so") \
-    .compute_cost(10) \
-    .kernel_name("mat_mul") \
-    .partial_flag(True) \
-    .need_check_supported(True) \
-    .attr("transpose_x1", "required", "bool", "all") \
-    .attr("transpose_x2", "required", "bool", "all") \
-    .attr("offset_x", "optional", "int", "all", "0") \
-    .input(0, "x1", False, "required", "all") \
-    .input(1, "x2", False, "required", "all") \
-    .input(2, "bias", False, "optional", "all") \
-    .input(3, "offset_w", False, "optional", "all") \
-    .output(0, "y", False, "required", "all") \
-    .dtype_format(DataType.I32_Default, DataType.I32_Default, DataType.I32_Default, DataType.I8_Default,
-                  DataType.I32_Default) \
-    .dtype_format(DataType.F16_FracNZ, DataType.F16_FracNZ, DataType.F16_Default, DataType.I8_Default,
-                  DataType.F16_FracNZ) \
-    .dtype_format(DataType.F16_FracNZ, DataType.F16_FracNZ, DataType.F32_Default, DataType.I8_Default,
-                  DataType.F32_FracNZ) \
-    .dtype_format(DataType.F32_NHWC, DataType.F32_NHWC, DataType.F32_NHWC, DataType.I8_Default,
-                  DataType.F32_NHWC) \
-    .dtype_format(DataType.F32_Default, DataType.F32_Default, DataType.F32_Default, DataType.I8_Default,
-                  DataType.F32_Default) \
-    .dtype_format(DataType.I32_NHWC, DataType.I32_NHWC, DataType.I32_NHWC, DataType.I8_Default,
-                  DataType.I32_NHWC) \
-    .get_op_info()
-
-
-@op_info_register(matmul_op_info)
-def _matmul_tbe():
-    """MatMul TBE register"""
-    return

mindspore/ops/_op_impl/tbe/matmul_ds.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.
-# ============================================================================
-
-"""MatMul op"""
-from mindspore.ops.op_info_register import op_info_register, TBERegOp, DataType
-
-matmul_op_info = TBERegOp("MatMul") \
-    .fusion_type("OPAQUE") \
-    .async_flag(False) \
-    .binfile_name("mat_mul.so") \
-    .compute_cost(10) \
-    .kernel_name("mat_mul") \
-    .partial_flag(True) \
-    .dynamic_shape(True) \
-    .attr("transpose_x1", "required", "bool", "all") \
-    .attr("transpose_x2", "required", "bool", "all") \
-    .attr("offset_x", "optional", "int", "all", "0") \
-    .input(0, "x1", False, "required", "all") \
-    .input(1, "x2", False, "required", "all") \
-    .input(2, "bias", False, "optional", "all") \
-    .input(3, "offset_w", False, "optional", "all") \
-    .output(0, "y", False, "required", "all") \
-    .dtype_format(DataType.F16_FracNZ, DataType.F16_FracNZ, DataType.F16_Default, DataType.I8_Default,
-                  DataType.F16_FracNZ) \
-    .dtype_format(DataType.F16_FracNZ, DataType.F16_FracNZ, DataType.F32_Default, DataType.I8_Default,
-                  DataType.F16_FracNZ) \
-    .dtype_format(DataType.F16_FracNZ, DataType.F16_FracNZ, DataType.F32_Default, DataType.I8_Default,
-                  DataType.F32_FracNZ) \
-    .get_op_info()
-
-
-@op_info_register(matmul_op_info)
-def _matmul_ds_tbe():
-    """MatMul TBE register"""
-    return

mindspore/ops/_op_impl/tbe/matmul_v2.py

@@ -1,50 +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.
-# ============================================================================
-
-"""MatMul op"""
-from mindspore.ops.op_info_register import op_info_register, TBERegOp, DataType
-
-matmul_v2_op_info = TBERegOp("MatMulV2") \
-    .fusion_type("OPAQUE") \
-    .async_flag(False) \
-    .binfile_name("mat_mul.so") \
-    .compute_cost(10) \
-    .kernel_name("mat_mul") \
-    .partial_flag(True) \
-    .need_check_supported(True) \
-    .attr("transpose_x1", "required", "bool", "all") \
-    .attr("transpose_x2", "required", "bool", "all") \
-    .attr("offset_x", "optional", "int", "all", "0") \
-    .input(0, "x1", False, "required", "all") \
-    .input(1, "x2", False, "required", "all") \
-    .input(2, "bias", False, "optional", "all") \
-    .input(3, "offset_w", False, "optional", "all") \
-    .output(0, "y", False, "required", "all") \
-    .is_dynamic_format(True) \
-    .dtype_format(DataType.I32_None, DataType.I32_None, DataType.I32_None, DataType.I8_None,
-                  DataType.I32_None) \
-    .dtype_format(DataType.F16_None, DataType.F16_None, DataType.F16_None, DataType.I8_None,
-                  DataType.F16_None) \
-    .dtype_format(DataType.F16_None, DataType.F16_None, DataType.F32_None, DataType.I8_None,
-                  DataType.F32_None) \
-    .dtype_format(DataType.F32_None, DataType.F32_None, DataType.F32_None, DataType.I8_None,
-                  DataType.F32_None) \
-    .get_op_info()
-
-
-@op_info_register(matmul_v2_op_info)
-def _matmul_v2_tbe():
-    """MatMulV2 TBE register"""
-    return

mindspore/ops/_op_impl/tbe/matrix_diag.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.
-# ============================================================================
-
-"""MatrixDiagD op"""
-from mindspore.ops.op_info_register import op_info_register, TBERegOp, DataType
-
-matrix_diag_d_op_info = TBERegOp("MatrixDiag") \
-    .fusion_type("ELEMWISE") \
-    .async_flag(False) \
-    .binfile_name("matrix_diag_d.so") \
-    .compute_cost(10) \
-    .kernel_name("matrix_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_5HD, DataType.F16_5HD, DataType.F16_5HD) \
-    .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_Default, DataType.F32_Default, DataType.F32_Default) \
-    .dtype_format(DataType.I32_5HD, DataType.I32_5HD, DataType.I32_5HD) \
-    .dtype_format(DataType.I32_Default, DataType.I32_Default, DataType.I32_Default) \
-    .dtype_format(DataType.I8_5HD, DataType.I8_5HD, DataType.I8_5HD) \
-    .dtype_format(DataType.I8_Default, DataType.I8_Default, DataType.I8_Default) \
-    .dtype_format(DataType.U8_5HD, DataType.U8_5HD, DataType.U8_5HD) \
-    .dtype_format(DataType.U8_Default, DataType.U8_Default, DataType.U8_Default) \
-    .get_op_info()
-
-
-@op_info_register(matrix_diag_d_op_info)
-def _matrix_diag_tbe():
-    """MatrixDiagD TBE register"""
-    return

mindspore/ops/_op_impl/tbe/matrix_diag_part.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.
-# ============================================================================
-
-"""MatrixDiagPartD op"""
-from mindspore.ops.op_info_register import op_info_register, TBERegOp, DataType
-
-matrix_diag_part_d_op_info = TBERegOp("MatrixDiagPart") \
-    .fusion_type("ELEMWISE") \
-    .async_flag(False) \
-    .binfile_name("matrix_diag_part_d.so") \
-    .compute_cost(10) \
-    .kernel_name("matrix_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_5HD, DataType.F16_5HD, DataType.F16_5HD) \
-    .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_Default, DataType.F32_Default, DataType.F32_Default) \
-    .dtype_format(DataType.I32_5HD, DataType.I32_5HD, DataType.I32_5HD) \
-    .dtype_format(DataType.I32_Default, DataType.I32_Default, DataType.I32_Default) \
-    .dtype_format(DataType.I8_5HD, DataType.I8_5HD, DataType.I8_5HD) \
-    .dtype_format(DataType.I8_Default, DataType.I8_Default, DataType.I8_Default) \
-    .dtype_format(DataType.U8_5HD, DataType.U8_5HD, DataType.U8_5HD) \
-    .dtype_format(DataType.U8_Default, DataType.U8_Default, DataType.U8_Default) \
-    .get_op_info()
-
-
-@op_info_register(matrix_diag_part_d_op_info)
-def _matrix_diag_part_tbe():
-    """MatrixDiagPartD TBE register"""
-    return

mindspore/ops/_op_impl/tbe/matrix_set_diag.py

@@ -1,46 +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.
-# ============================================================================
-
-"""MatrixSetDiagD op"""
-from mindspore.ops.op_info_register import op_info_register, TBERegOp, DataType
-
-matrix_diag_d_op_info = TBERegOp("MatrixSetDiag") \
-    .fusion_type("ELEMWISE") \
-    .async_flag(False) \
-    .binfile_name("matrix_diag_d.so") \
-    .compute_cost(10) \
-    .kernel_name("matrix_set_diag_d") \
-    .partial_flag(True) \
-    .input(0, "x", False, "required", "all") \
-    .input(1, "diagonal", False, "required", "all") \
-    .input(2, "assist", False, "required", "all") \
-    .output(0, "y", False, "required", "all") \
-    .dtype_format(DataType.F16_5HD, DataType.F16_5HD, DataType.F16_5HD, DataType.F16_5HD) \
-    .dtype_format(DataType.F16_Default, DataType.F16_Default, DataType.F16_Default, DataType.F16_Default) \
-    .dtype_format(DataType.F32_5HD, DataType.F32_5HD, DataType.F32_5HD, DataType.F32_5HD) \
-    .dtype_format(DataType.F32_Default, DataType.F32_Default, DataType.F32_Default, DataType.F32_Default) \
-    .dtype_format(DataType.I32_5HD, DataType.I32_5HD, DataType.I32_5HD, DataType.I32_5HD) \
-    .dtype_format(DataType.I32_Default, DataType.I32_Default, DataType.I32_Default, DataType.I32_Default) \
-    .dtype_format(DataType.I8_5HD, DataType.I8_5HD, DataType.I8_5HD, DataType.I8_5HD) \
-    .dtype_format(DataType.I8_Default, DataType.I8_Default, DataType.I8_Default, DataType.I8_Default) \
-    .dtype_format(DataType.U8_5HD, DataType.U8_5HD, DataType.U8_5HD, DataType.U8_5HD) \
-    .dtype_format(DataType.U8_Default, DataType.U8_Default, DataType.U8_Default, DataType.U8_Default) \
-    .get_op_info()
-
-
-@op_info_register(matrix_diag_d_op_info)
-def _matrix_set_diag_tbe():
-    """MatrixSetDiagD TBE register"""
-    return

mindspore/ops/_op_impl/tbe/max_pool.py

@@ -1,39 +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.
-# ============================================================================
-
-"""MaxPool op"""
-from mindspore.ops.op_info_register import op_info_register, TBERegOp, DataType
-
-max_pool_op_info = TBERegOp("MaxPool") \
-    .fusion_type("OPAQUE") \
-    .async_flag(False) \
-    .binfile_name("max_pool.so") \
-    .compute_cost(10) \
-    .kernel_name("max_pool") \
-    .partial_flag(True) \
-    .attr("kernel_size", "required", "listInt", "all") \
-    .attr("strides", "required", "listInt", "all") \
-    .attr("pad_mode", "required", "str", "all") \
-    .attr("format", "required", "str", "all") \
-    .input(0, "input_data", False, "required", "all") \
-    .output(0, "output_data", False, "required", "all") \
-    .dtype_format(DataType.F16_5HD, DataType.F16_5HD) \
-    .get_op_info()
-
-
-@op_info_register(max_pool_op_info)
-def _max_pool_tbe():
-    """MaxPool TBE register"""
-    return

mindspore/ops/_op_impl/tbe/max_pool3d.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.
-# ============================================================================
-
-
-"""MaxPool3D op"""
-from mindspore.ops.op_info_register import op_info_register, TBERegOp, DataType
-
-max_pool3d_op_info = TBERegOp("MaxPool3D") \
-    .fusion_type("OPAQUE") \
-    .async_flag(False) \
-    .binfile_name("max_pool3d.so") \
-    .compute_cost(10) \
-    .kernel_name("max_pool3d") \
-    .partial_flag(True) \
-    .attr("kernel_size", "required", "listInt", "all") \
-    .attr("strides", "required", "listInt", "all") \
-    .attr("pad_mode", "required", "str", "all") \
-    .attr("pad_list", "optional", "listInt", "all", "0,0,0") \
-    .attr("dilation", "optional", "listInt", "all", "1,1,1") \
-    .attr("ceil_mode", "optional", "int", "all", "0") \
-    .attr("format", "optional", "str", "all") \
-    .input(0, "x", False, "required", "all") \
-    .output(0, "y", False, "required", "all") \
-    .is_dynamic_format(True) \
-    .dtype_format(DataType.None_None, DataType.None_None) \
-    .get_op_info()
-
-
-@op_info_register(max_pool3d_op_info)
-def _max_pool_3d_tbe():
-    """MaxPool3D TBE register"""
-    return

mindspore/ops/_op_impl/tbe/max_pool3d_grad.py

@@ -1,43 +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.
-# ============================================================================
-
-
-"""MaxPool3DGrad op"""
-from mindspore.ops.op_info_register import op_info_register, TBERegOp, DataType
-
-max_pool3d_grad_op_info = TBERegOp("MaxPool3DGrad") \
-    .fusion_type("OPAQUE") \
-    .async_flag(False) \
-    .binfile_name("max_pool3d_grad.so") \
-    .compute_cost(10) \
-    .kernel_name("max_pool3d_grad") \
-    .partial_flag(True) \
-    .attr("kernel_size", "required", "listInt", "all") \
-    .attr("strides", "required", "listInt", "all") \
-    .attr("pad_mode", "optional", "str", "all") \
-    .attr("pad_list", "required", "listInt", "all", "0,0,0") \
-    .attr("format", "optional", "str", "all") \
-    .input(0, "orig_x", False, "required", "all") \
-    .input(1, "orig_y", False, "required", "all") \
-    .input(2, "grads", False, "required", "all") \
-    .output(0, "y", False, "required", "all") \
-    .dtype_format(DataType.F16_NDC1HWC0, DataType.F16_NDC1HWC0, DataType.F16_NDC1HWC0, DataType.F32_NDC1HWC0) \
-    .get_op_info()
-
-
-@op_info_register(max_pool3d_grad_op_info)
-def _max_pool_3d_grad_tbe():
-    """MaxPool3DGrad TBE register"""
-    return

mindspore/ops/_op_impl/tbe/max_pool3d_grad_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.
-# ============================================================================
-
-
-"""MaxPool3DGradGrad op"""
-from mindspore.ops.op_info_register import op_info_register, TBERegOp, DataType
-
-max_pool3d_grad_grad_op_info = TBERegOp("MaxPool3DGradGrad") \
-    .fusion_type("OPAQUE") \
-    .async_flag(False) \
-    .binfile_name("max_pool3d_grad_grad_d.so") \
-    .compute_cost(10) \
-    .kernel_name("max_pool3d_grad_grad_d") \
-    .partial_flag(True) \
-    .attr("kernel_size", "required", "listInt", "all") \
-    .attr("strides", "required", "listInt", "all") \
-    .attr("pad_list", "required", "listInt", "all") \
-    .attr("format", "optional", "str", "all") \
-    .input(0, "orig_in", False, "required", "all") \
-    .input(1, "orig_out", False, "required", "all") \
-    .input(2, "grads", False, "required", "all") \
-    .input(3, "assist", False, "required", "all") \
-    .output(0, "y", False, "required", "all") \
-    .dtype_format(DataType.F16_NDC1HWC0, DataType.F16_NDC1HWC0, DataType.F16_NDC1HWC0, DataType.F16_NDC1HWC0,
-                  DataType.F16_NDC1HWC0) \
-    .get_op_info()
-
-
-@op_info_register(max_pool3d_grad_grad_op_info)
-def _max_pool_3d_grad_grad_tbe():
-    """MaxPool3DGradGrad TBE register"""
-    return

mindspore/ops/_op_impl/tbe/max_pool_ds.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.
-# ============================================================================
-
-"""MaxPool op"""
-from mindspore.ops.op_info_register import op_info_register, TBERegOp, DataType
-
-max_pool_op_info = TBERegOp("MaxPool") \
-    .fusion_type("OPAQUE") \
-    .async_flag(False) \
-    .binfile_name("max_pool.so") \
-    .compute_cost(10) \
-    .kernel_name("max_pool") \
-    .partial_flag(True) \
-    .dynamic_shape(True) \
-    .attr("kernel_size", "required", "listInt", "all") \
-    .attr("strides", "required", "listInt", "all") \
-    .attr("pad_mode", "required", "str", "all") \
-    .attr("format", "required", "str", "all") \
-    .input(0, "input_data", False, "required", "all") \
-    .output(0, "output_data", False, "required", "all") \
-    .dtype_format(DataType.F16_5HD, DataType.F16_5HD) \
-    .get_op_info()
-
-
-@op_info_register(max_pool_op_info)
-def _max_pool_ds_tbe():
-    """MaxPool TBE register"""
-    return

mindspore/ops/_op_impl/tbe/max_pool_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.
-# ============================================================================
-
-"""MaxPoolGrad op"""
-from mindspore.ops.op_info_register import op_info_register, TBERegOp, DataType
-
-max_pool_grad_op_info = TBERegOp("MaxPoolGrad") \
-    .fusion_type("OPAQUE") \
-    .async_flag(False) \
-    .binfile_name("max_pool_grad.so") \
-    .compute_cost(10) \
-    .kernel_name("max_pool_grad") \
-    .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", "NHWC") \
-    .input(0, "x1", False, "required", "all") \
-    .input(1, "x2", False, "required", "all") \
-    .input(2, "grad", False, "required", "all") \
-    .output(0, "y", False, "required", "all") \
-    .dtype_format(DataType.None_None, DataType.None_None, DataType.None_None, DataType.None_None) \
-    .dynamic_shape(True) \
-    .is_dynamic_format(True) \
-    .get_op_info()
-
-
-@op_info_register(max_pool_grad_op_info)
-def _max_pool_grad_tbe():
-    """MaxPoolGrad TBE register"""
-    return

mindspore/ops/_op_impl/tbe/max_pool_grad_grad.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.
-# ============================================================================
-
-"""MaxPoolGradGrad op"""
-from mindspore.ops.op_info_register import op_info_register, TBERegOp, DataType
-
-max_pool_grad_grad_op_info = TBERegOp("MaxPoolGradGrad") \
-    .fusion_type("OPAQUE") \
-    .async_flag(False) \
-    .binfile_name("max_pool_grad_grad.so") \
-    .compute_cost(10) \
-    .kernel_name("max_pool_grad_grad") \
-    .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") \
-    .input(0, "x1", False, "required", "all") \
-    .input(1, "x2", False, "required", "all") \
-    .input(2, "grad", False, "required", "all") \
-    .output(0, "y", False, "required", "all") \
-    .dtype_format(DataType.F16_5HD, DataType.F16_5HD, DataType.F16_5HD, DataType.F16_5HD) \
-    .get_op_info()
-
-
-@op_info_register(max_pool_grad_grad_op_info)
-def _max_pool_grad_grad_tbe():
-    """MaxPoolGradGrad TBE register"""
-    return

mindspore/ops/_op_impl/tbe/max_pool_grad_grad_with_argmax.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.
-# ============================================================================
-
-"""MaxPoolGradGradWithArgmax op"""
-from mindspore.ops.op_info_register import op_info_register, TBERegOp, DataType
-
-max_pool_grad_grad_with_argmax_op_info = TBERegOp("MaxPoolGradGradWithArgmax") \
-    .fusion_type("OPAQUE") \
-    .async_flag(False) \
-    .binfile_name("max_pool_grad_grad_with_argmax.so") \
-    .compute_cost(10) \
-    .kernel_name("max_pool_grad_grad_with_argmax") \
-    .partial_flag(True) \
-    .attr("kernel_size", "required", "listInt", "all") \
-    .attr("strides", "required", "listInt", "all") \
-    .attr("pad_mode", "required", "str", "all") \
-    .input(0, "x", False, "required", "all") \
-    .input(1, "grad", False, "required", "all") \
-    .input(2, "argmax", False, "optional", "all") \
-    .output(0, "y", False, "required", "all") \
-    .dtype_format(DataType.F16_5HD, DataType.F16_5HD, DataType.U16_5HD, DataType.F16_5HD) \
-    .dtype_format(DataType.F16_5HD, DataType.F16_5HD, DataType.I64_5HD, DataType.F16_5HD) \
-    .get_op_info()
-
-
-@op_info_register(max_pool_grad_grad_with_argmax_op_info)
-def _max_pool_grad_grad_with_argmax_tbe():
-    """MaxPoolGradGradWithArgmax TBE register"""
-    return