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

mindspore/ops/_op_impl/tbe/basic_lstm_cell_c_state_grad_v2.py

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

mindspore/ops/_op_impl/tbe/basic_lstm_cell_input_grad.py

@@ -1,42 +0,0 @@
-# Copyright 2020 Huawei Technologies Co., Ltd
-#
-# Licensed under the Apache License, Version 2.0 (the "License");
-# you may not use this file except in compliance with the License.
-# You may obtain a copy of the License at
-#
-# http://www.apache.org/licenses/LICENSE-2.0
-#
-# Unless required by applicable law or agreed to in writing, software
-# distributed under the License is distributed on an "AS IS" BASIS,
-# WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
-# See the License for the specific language governing permissions and
-# limitations under the License.
-# ============================================================================
-
-"""BasicLSTMCellInputGrad op"""
-from mindspore.ops.op_info_register import op_info_register, TBERegOp, DataType
-
-basic_lstm_cell_input_grad_op_info = TBERegOp("BasicLSTMCellInputGrad") \
-    .fusion_type("OPAQUE") \
-    .async_flag(False) \
-    .binfile_name("basic_lstm_cell_input_grad.so") \
-    .compute_cost(10) \
-    .kernel_name("basic_lstm_cell_input_grad") \
-    .attr("keep_prob", "optional", "float", "all", "1.0") \
-    .partial_flag(True) \
-    .input(0, "dgate", False, "required", "all") \
-    .input(1, "w", False, "required", "all", reshape_type="NC") \
-    .input(2, "dropout_mask", False, "optional", "all") \
-    .output(0, "dxt", False, "required", "all") \
-    .output(1, "dht", False, "required", "all") \
-    .dtype_format(DataType.F16_FracNZ, DataType.F16_FracZ, DataType.U8_Default, DataType.F32_FracNZ,
-                  DataType.F32_FracNZ) \
-    .dtype_format(DataType.F16_FracNZ, DataType.F16_FracZ, DataType.U8_Default, DataType.F16_FracNZ,
-                  DataType.F16_FracNZ) \
-    .get_op_info()
-
-
-@op_info_register(basic_lstm_cell_input_grad_op_info)
-def _basic_lstm_cell_input_grad_tbe():
-    """BasicLSTMCellInputGrad TBE register"""
-    return

mindspore/ops/_op_impl/tbe/basic_lstm_cell_weight_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.
-# ============================================================================
-
-"""BasicLSTMCellWeightGrad op"""
-from mindspore.ops.op_info_register import op_info_register, TBERegOp, DataType
-
-basic_lstm_cell_weight_grad_op_info = TBERegOp("BasicLSTMCellWeightGrad") \
-    .fusion_type("OPAQUE") \
-    .async_flag(False) \
-    .binfile_name("basic_lstm_cell_weight_grad.so") \
-    .compute_cost(10) \
-    .kernel_name("basic_lstm_cell_weight_grad") \
-    .partial_flag(True) \
-    .input(0, "x", False, "required", "all") \
-    .input(1, "h", False, "required", "all") \
-    .input(2, "dgate", False, "required", "all") \
-    .output(0, "dw", False, "required", "all", reshape_type="CN") \
-    .output(1, "db", False, "required", "all") \
-    .dtype_format(DataType.F16_FracNZ, DataType.F16_FracNZ, DataType.F16_FracNZ, DataType.F16_FracZ,
-                  DataType.F32_Default) \
-    .dtype_format(DataType.F16_FracNZ, DataType.F16_FracNZ, DataType.F16_FracNZ, DataType.F16_FracZ,
-                  DataType.F16_Default) \
-    .get_op_info()
-
-
-@op_info_register(basic_lstm_cell_weight_grad_op_info)
-def _basic_lstm_cell_weight_grad_tbe():
-    """BasicLSTMCellWeightGrad TBE register"""
-    return

mindspore/ops/_op_impl/tbe/batch_matmul.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.
-# ============================================================================
-
-"""BatchMatMul op"""
-from mindspore.ops.op_info_register import op_info_register, TBERegOp, DataType
-
-batch_matmul_op_info = TBERegOp("BatchMatMul") \
-    .fusion_type("OPAQUE") \
-    .async_flag(False) \
-    .binfile_name("batch_matmul.so") \
-    .compute_cost(10) \
-    .kernel_name("batch_matmul") \
-    .attr("transpose_x1", "required", "bool", "all") \
-    .attr("transpose_x2", "required", "bool", "all") \
-    .partial_flag(True) \
-    .input(0, "x1", False, "required", "all") \
-    .input(1, "x2", False, "required", "all") \
-    .input(2, "bias", 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.I32_None) \
-    .dtype_format(DataType.F16_None, DataType.F16_None, DataType.F16_None, DataType.F16_None) \
-    .dtype_format(DataType.F32_None, DataType.F32_None, DataType.F32_None, DataType.F32_None) \
-    .get_op_info()
-
-
-@op_info_register(batch_matmul_op_info)
-def _batch_matmul_tbe():
-    """BatchMatMul TBE register"""
-    return

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

mindspore/ops/_op_impl/tbe/batch_matmul_v2.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.
-# ============================================================================
-
-"""BatchMatMul op"""
-from mindspore.ops.op_info_register import op_info_register, TBERegOp, DataType
-
-batch_matmul_v2_op_info = TBERegOp("BatchMatMulV2") \
-    .fusion_type("OPAQUE") \
-    .async_flag(False) \
-    .binfile_name("batch_matmul_v2.so") \
-    .compute_cost(10) \
-    .kernel_name("batch_matmul_v2") \
-    .attr("transpose_x1", "required", "bool", "all") \
-    .attr("transpose_x2", "required", "bool", "all") \
-    .attr("offset_x", "optional", "int", "all", "0") \
-    .partial_flag(True) \
-    .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.I32_None,
-                  DataType.I32_None) \
-    .dtype_format(DataType.F16_None, DataType.F16_None, DataType.F16_None, DataType.F16_None,
-                  DataType.F16_None) \
-    .dtype_format(DataType.F32_None, DataType.F32_None, DataType.F32_None, DataType.F32_None,
-                  DataType.F32_None) \
-    .get_op_info()
-
-
-@op_info_register(batch_matmul_v2_op_info)
-def _batch_matmul_v2_tbe():
-    """BatchMatMulV2 TBE register"""
-    return

mindspore/ops/_op_impl/tbe/batch_to_space.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.
-# ============================================================================
-
-"""BatchToSpace op"""
-from mindspore.ops.op_info_register import op_info_register, TBERegOp, DataType
-
-batch_to_space_op_info = TBERegOp("BatchToSpaceD") \
-    .fusion_type("OPAQUE") \
-    .async_flag(False) \
-    .binfile_name("batch_to_space_d.so") \
-    .compute_cost(10) \
-    .kernel_name("batch_to_space_d") \
-    .partial_flag(True) \
-    .attr("block_size", "required", "int", "all") \
-    .attr("crops", "required", "listListInt", "all", "[[]]") \
-    .input(0, "x", False, "required", "all") \
-    .output(0, "y", False, "required", "all") \
-    .dtype_format(DataType.F16_5HD, DataType.F16_5HD) \
-    .dtype_format(DataType.F32_5HD, DataType.F32_5HD) \
-    .get_op_info()
-
-
-@op_info_register(batch_to_space_op_info)
-def _batch_to_space_tbe():
-    """BatchToSpace TBE register"""
-    return

mindspore/ops/_op_impl/tbe/batch_to_space_nd.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.
-# ============================================================================
-
-"""BatchToSpaceND op"""
-from mindspore.ops.op_info_register import op_info_register, TBERegOp, DataType
-
-batch_to_space_nd_op_info = TBERegOp("BatchToSpaceNDD") \
-    .fusion_type("OPAQUE") \
-    .async_flag(False) \
-    .binfile_name("batch_to_space_nd_d.so") \
-    .compute_cost(10) \
-    .kernel_name("batch_to_space_nd_d") \
-    .partial_flag(True) \
-    .attr("block_shape", "required", "listInt", "all", "[]") \
-    .attr("crops", "required", "listListInt", "all", "[[]]") \
-    .input(0, "x", False, "required", "all", reshape_type="NH") \
-    .output(0, "y", False, "required", "all", reshape_type="NH") \
-    .dtype_format(DataType.F16_5HD, DataType.F16_5HD) \
-    .dtype_format(DataType.F32_5HD, DataType.F32_5HD) \
-    .get_op_info()
-
-
-@op_info_register(batch_to_space_nd_op_info)
-def _batch_to_space_nd_tbe():
-    """BatchToSpaceND TBE register"""
-    return

mindspore/ops/_op_impl/tbe/batch_to_space_nd_ds.py

@@ -1,39 +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.
-# ============================================================================
-
-"""BatchToSpaceND op"""
-from mindspore.ops.op_info_register import op_info_register, TBERegOp, DataType
-
-batch_to_space_nd_ds_op_info = TBERegOp("BatchToSpaceND") \
-    .fusion_type("OPAQUE") \
-    .async_flag(False) \
-    .binfile_name("batch_to_space_nd_d.so") \
-    .compute_cost(10) \
-    .kernel_name("batch_to_space_nd_d") \
-    .partial_flag(True) \
-    .dynamic_shape(True)\
-    .attr("block_shape", "required", "listInt", "all") \
-    .attr("crops", "required", "listListInt", "all") \
-    .input(0, "x", False, "required", "all", reshape_type="NH") \
-    .output(0, "y", False, "required", "all", reshape_type="NH") \
-    .dtype_format(DataType.F16_5HD, DataType.F16_5HD) \
-    .dtype_format(DataType.F32_5HD, DataType.F32_5HD) \
-    .get_op_info()
-
-
-@op_info_register(batch_to_space_nd_ds_op_info)
-def _batch_to_space_nd_ds_tbe():
-    """BatchToSpaceND TBE register"""
-    return

mindspore/ops/_op_impl/tbe/batch_to_space_nd_v2.py

@@ -1,41 +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.
-# ============================================================================
-
-"""BatchToSpaceNDV2 op"""
-from mindspore.ops.op_info_register import op_info_register, TBERegOp, DataType
-
-batch_to_space_nd_v2_op_info = TBERegOp("BatchToSpaceNDV2") \
-    .fusion_type("OPAQUE") \
-    .async_flag(False) \
-    .binfile_name("batch_to_space_nd.so") \
-    .compute_cost(10) \
-    .kernel_name("batch_to_space_nd") \
-    .partial_flag(True) \
-    .need_check_supported(True) \
-    .dynamic_compile_static(True) \
-    .dynamic_shape(True) \
-    .input(0, "x", False, "required", "all", reshape_type="NHC") \
-    .input(1, "block_shape", False, "required", "all", "optional") \
-    .input(2, "crops", False, "required", "all", "optional") \
-    .output(0, "y", False, "required", "all", reshape_type="NHC") \
-    .is_dynamic_format(True) \
-    .dtype_format(DataType.None_None, DataType.None_None, DataType.None_None, DataType.None_None) \
-    .get_op_info()
-
-
-@op_info_register(batch_to_space_nd_v2_op_info)
-def _batch_to_space_nd_v2_tbe():
-    """BatchToSpaceND TBE register"""
-    return

mindspore/ops/_op_impl/tbe/batchnorm.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.
-# ============================================================================
-
-"""BatchNorm op"""
-from mindspore.ops.op_info_register import op_info_register, TBERegOp, DataType
-
-batch_norm_op_info = TBERegOp("BatchNorm") \
-    .fusion_type("OPAQUE") \
-    .async_flag(False) \
-    .binfile_name("batch_norm.so") \
-    .compute_cost(10) \
-    .kernel_name("batch_norm") \
-    .partial_flag(True) \
-    .attr("epsilon", "optional", "float", "all") \
-    .attr("format", "optional", "str", "all") \
-    .attr("is_training", "optional", "bool", "all") \
-    .input(0, "x", False, "required", "all") \
-    .input(1, "scale", False, "required", "all") \
-    .input(2, "offset", False, "required", "all") \
-    .input(3, "mean", False, "optional", "all") \
-    .input(4, "variance", False, "optional", "all") \
-    .output(0, "y", False, "required", "all") \
-    .output(1, "batch_mean", False, "required", "all") \
-    .output(2, "batch_variance", False, "required", "all") \
-    .output(3, "reserve_space_1", False, "optional", "all") \
-    .output(4, "reserve_space_2", False, "optional", "all") \
-    .output(5, "reserve_space_3", False, "optional", "all") \
-    .dtype_format(DataType.F16_Default, DataType.F32_Default, DataType.F32_Default, DataType.F32_Default,
-                  DataType.F32_Default, DataType.F16_Default, DataType.F32_Default, DataType.F32_Default,
-                  DataType.F32_Default, DataType.F32_Default, DataType.F32_Default) \
-    .dtype_format(DataType.F16_5HD, DataType.F32_5HD, DataType.F32_5HD, DataType.F32_5HD,
-                  DataType.F32_5HD, DataType.F16_5HD, DataType.F32_5HD, 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,
-                  DataType.F32_Default, DataType.F32_Default, DataType.F32_Default, DataType.F32_Default,
-                  DataType.F32_Default, DataType.F32_Default, DataType.F32_Default) \
-    .dtype_format(DataType.F32_5HD, DataType.F32_5HD, DataType.F32_5HD, DataType.F32_5HD,
-                  DataType.F32_5HD, DataType.F32_5HD, DataType.F32_5HD, DataType.F32_5HD,
-                  DataType.F32_5HD, DataType.F32_5HD, DataType.F32_5HD) \
-    .get_op_info()
-
-
-@op_info_register(batch_norm_op_info)
-def _batch_norm_tbe():
-    """BatchNorm TBE register"""
-    return

mindspore/ops/_op_impl/tbe/batchnorm_grad.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.
-# ============================================================================
-
-"""BatchNormGrad op"""
-from mindspore.ops.op_info_register import op_info_register, TBERegOp, DataType
-
-batch_norm_grad_op_info = TBERegOp("BatchNormGrad") \
-    .fusion_type("OPAQUE") \
-    .async_flag(False) \
-    .binfile_name("batch_norm_grad.so") \
-    .compute_cost(10) \
-    .kernel_name("batch_norm_grad") \
-    .partial_flag(True) \
-    .attr("epsilon", "optional", "float", "all") \
-    .attr("format", "optional", "str", "all") \
-    .attr("is_training", "optional", "bool", "all") \
-    .input(0, "y_backprop", False, "required", "all") \
-    .input(1, "x", False, "required", "all") \
-    .input(2, "scale", False, "required", "all") \
-    .input(3, "reserve_space_1", False, "required", "all") \
-    .input(4, "reserve_space_2", False, "required", "all") \
-    .input(5, "reserve_space_3", False, "required", "all") \
-    .output(0, "x_backprop", False, "required", "all") \
-    .output(1, "scale_backprop", False, "required", "all") \
-    .output(2, "offset_backprop", False, "required", "all") \
-    .output(3, "reserve_space_4", False, "optional", "all") \
-    .output(4, "reserve_space_5", False, "optional", "all") \
-    .dtype_format(DataType.F16_Default, DataType.F16_Default, DataType.F32_Default, DataType.F32_Default,
-                  DataType.F32_Default, DataType.F32_Default, DataType.F16_Default, DataType.F32_Default,
-                  DataType.F32_Default, DataType.F32_Default, DataType.F32_Default) \
-    .dtype_format(DataType.F16_5HD, DataType.F16_5HD, DataType.F32_5HD, DataType.F32_5HD,
-                  DataType.F32_5HD, DataType.F32_5HD, DataType.F16_5HD, 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,
-                  DataType.F32_Default, DataType.F32_Default, DataType.F32_Default, DataType.F32_Default,
-                  DataType.F32_Default, DataType.F32_Default, DataType.F32_Default) \
-    .dtype_format(DataType.F32_5HD, DataType.F32_5HD, DataType.F32_5HD, DataType.F32_5HD,
-                  DataType.F32_5HD, DataType.F32_5HD, DataType.F32_5HD, DataType.F32_5HD,
-                  DataType.F32_5HD, DataType.F32_5HD, DataType.F32_5HD) \
-    .get_op_info()
-
-
-@op_info_register(batch_norm_grad_op_info)
-def _batch_norm_grad_tbe():
-    """BatchNormGrad TBE register"""
-    return

mindspore/ops/_op_impl/tbe/bce_with_logits_loss.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 unde:q!r 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.
-# ============================================================================
-
-"""BCEWithLogitsLoss op"""
-from mindspore.ops.op_info_register import op_info_register, TBERegOp, DataType
-
-bce_with_logits_loss_op_info = TBERegOp("BCEWithLogitsLoss") \
-    .fusion_type("OPAQUE") \
-    .async_flag(False) \
-    .binfile_name("sigmoid_cross_entropy_with_logits_v2.so") \
-    .compute_cost(10) \
-    .kernel_name("sigmoid_cross_entropy_with_logits_v2") \
-    .partial_flag(True) \
-    .dynamic_shape(True) \
-    .is_dynamic_format(True) \
-    .attr("reduction", "optional", "str", "all", "mean") \
-    .input(0, "predict", False, "required", "all") \
-    .input(1, "target", False, "required", "all") \
-    .input(2, "weight", False, "optional", "all") \
-    .input(3, "pos_weight", False, "optional", "all") \
-    .output(0, "loss", False, "required", "all") \
-    .dtype_format(DataType.None_None, DataType.None_None, DataType.None_None, DataType.None_None,
-                  DataType.None_None) \
-    .get_op_info()
-
-
-@op_info_register(bce_with_logits_loss_op_info)
-def _bce_with_logits_loss_op_tbe():
-    """BCEWithLogitsLoss TBE register"""
-    return

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