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

mindspore/ops/_op_impl/tbe/bn_training_update_v3.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.
-# ============================================================================
-
-"""BNTrainingUpdateV3 op"""
-from mindspore.ops.op_info_register import op_info_register, TBERegOp, DataType
-
-bn_training_update_v3_op_info = TBERegOp("BNTrainingUpdateV3") \
-    .fusion_type("OPAQUE") \
-    .async_flag(False) \
-    .binfile_name("bn_training_update_v3.so") \
-    .compute_cost(10) \
-    .kernel_name("bn_training_update_v3") \
-    .partial_flag(True) \
-    .attr("epsilon", "required", "float", "all") \
-    .input(0, "x", False, "required", "all", reshape_type="NC") \
-    .input(1, "sum", False, "required", "all") \
-    .input(2, "square_sum", False, "required", "all") \
-    .input(3, "scale", False, "required", "all") \
-    .input(4, "offset", False, "required", "all") \
-    .output(0, "y", False, "required", "all", reshape_type="NC") \
-    .output(1, "batch_mean", False, "required", "all") \
-    .output(2, "batch_variance", False, "required", "all") \
-    .output(3, "reserve_1", False, "required", "all") \
-    .output(4, "reserve_2", False, "required", "all") \
-    .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) \
-    .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) \
-    .get_op_info()
-
-
-@op_info_register(bn_training_update_v3_op_info)
-def _bn_training_update_v3_tbe():
-    """BNTrainingUpdateV3 TBE register"""
-    return

mindspore/ops/_op_impl/tbe/bounding_box_decode.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.
-# ============================================================================
-
-"""BoundingBoxDecode op"""
-from mindspore.ops.op_info_register import op_info_register, TBERegOp, DataType
-
-bounding_box_decode_op_info = TBERegOp("BoundingBoxDecode") \
-    .fusion_type("OPAQUE") \
-    .async_flag(False) \
-    .binfile_name("bounding_box_decode.so") \
-    .compute_cost(10) \
-    .kernel_name("bounding_box_decode") \
-    .partial_flag(True) \
-    .attr("means", "optional", "listFloat", "all") \
-    .attr("stds", "optional", "listFloat", "all") \
-    .attr("max_shape", "optional", "listInt", "all", "None") \
-    .attr("wh_ratio_clip", "optional", "float", "all", "0.016") \
-    .input(0, "rois", False, "required", "all") \
-    .input(1, "deltas", False, "required", "all") \
-    .output(0, "bboxes", 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) \
-    .get_op_info()
-
-
-@op_info_register(bounding_box_decode_op_info)
-def _bounding_box_decode_tbe():
-    """BoundingBoxDecode TBE register"""
-    return

mindspore/ops/_op_impl/tbe/bounding_box_decode_ds.py

@@ -1,42 +0,0 @@
-# Copyright 2021 Huawei Technologies Co., Ltd
-#
-# Licensed under the Apache License, Version 2.0 (the "License");
-# you may not use this file except in compliance with the License.
-# You may obtain a copy of the License at
-#
-# http://www.apache.org/licenses/LICENSE-2.0
-#
-# Unless required by applicable law or agreed to in writing, software
-# distributed under the License is distributed on an "AS IS" BASIS,
-# WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
-# See the License for the specific language governing permissions and
-# limitations under the License.
-# ============================================================================
-
-"""BoundingBoxDecode op"""
-from mindspore.ops.op_info_register import op_info_register, TBERegOp, DataType
-
-bounding_box_decode_op_info = TBERegOp("BoundingBoxDecode") \
-    .fusion_type("OPAQUE") \
-    .async_flag(False) \
-    .binfile_name("bounding_box_decode.so") \
-    .compute_cost(10) \
-    .kernel_name("bounding_box_decode") \
-    .partial_flag(True) \
-    .dynamic_shape(True) \
-    .attr("means", "optional", "listFloat", "all") \
-    .attr("stds", "optional", "listFloat", "all") \
-    .attr("max_shape", "optional", "listInt", "all", "None") \
-    .attr("wh_ratio_clip", "optional", "float", "all", "0.016") \
-    .input(0, "rois", False, "required", "all") \
-    .input(1, "deltas", False, "required", "all") \
-    .output(0, "bboxes", 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) \
-    .get_op_info()
-
-
-@op_info_register(bounding_box_decode_op_info)
-def _bounding_box_decode_ds_tbe():
-    """BoundingBoxDecode TBE register"""
-    return

mindspore/ops/_op_impl/tbe/bounding_box_encode.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.
-# ============================================================================
-
-"""BoundingBoxEncode op"""
-from mindspore.ops.op_info_register import op_info_register, TBERegOp, DataType
-
-bounding_box_encode_op_info = TBERegOp("BoundingBoxEncode") \
-    .fusion_type("OPAQUE") \
-    .async_flag(False) \
-    .binfile_name("bounding_box_encode.so") \
-    .compute_cost(10) \
-    .kernel_name("bounding_box_encode") \
-    .partial_flag(True) \
-    .attr("means", "optional", "listFloat", "all") \
-    .attr("stds", "optional", "listFloat", "all") \
-    .input(0, "anchor_box", False, "required", "all") \
-    .input(1, "ground_truth_box", False, "required", "all") \
-    .output(0, "delats", False, "required", "all") \
-    .dtype_format(DataType.F16_Default, DataType.F16_Default, DataType.F16_Default) \
-    .get_op_info()
-
-
-@op_info_register(bounding_box_encode_op_info)
-def _bounding_box_encode_tbe():
-    """BoundingBoxEncode TBE register"""
-    return

mindspore/ops/_op_impl/tbe/broadcast_to.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.
-# ============================================================================
-
-"""BroadcastTo op"""
-from mindspore.ops.op_info_register import op_info_register, TBERegOp, DataType
-
-broadcast_to_op_info = TBERegOp("BroadcastTo") \
-    .fusion_type("OPAQUE") \
-    .async_flag(False) \
-    .binfile_name("broadcast_to_d.so") \
-    .compute_cost(10) \
-    .kernel_name("broadcast_to_d") \
-    .partial_flag(True) \
-    .attr("shape", "required", "listInt", "all", "[]") \
-    .input(0, "x", False, "required", "all") \
-    .output(0, "y", False, "required", "all") \
-    .dtype_format(DataType.F16_Default, DataType.F16_Default) \
-    .dtype_format(DataType.F32_Default, DataType.F32_Default) \
-    .dtype_format(DataType.I32_Default, DataType.I32_Default) \
-    .dtype_format(DataType.I8_Default, DataType.I8_Default) \
-    .dtype_format(DataType.U8_Default, DataType.U8_Default) \
-    .get_op_info()
-
-
-@op_info_register(broadcast_to_op_info)
-def _broadcast_to_tbe():
-    """BroadcastTo TBE register"""
-    return

mindspore/ops/_op_impl/tbe/broadcast_to_ds.py

@@ -1,44 +0,0 @@
-# Copyright 2021 Huawei Technologies Co., Ltd
-#
-# Licensed under the Apache License, Version 2.0 (the "License");
-# you may not use this file except in compliance with the License.
-# You may obtain a copy of the License at
-#
-# http://www.apache.org/licenses/LICENSE-2.0
-#
-# Unless required by applicable law or agreed to in writing, software
-# distributed under the License is distributed on an "AS IS" BASIS,
-# WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
-# See the License for the specific language governing permissions and
-# limitations under the License.
-# ============================================================================
-
-"""BroadcastTo op"""
-from mindspore.ops.op_info_register import op_info_register, TBERegOp, DataType
-
-broadcast_to_op_info = TBERegOp("DynamicBroadcastTo") \
-    .fusion_type("OPAQUE") \
-    .async_flag(False) \
-    .binfile_name("broadcast_to.so") \
-    .compute_cost(10) \
-    .kernel_name("broadcast_to") \
-    .partial_flag(True) \
-    .dynamic_shape(True) \
-    .input(0, "x", False, "required", "all") \
-    .input(1, "shape", 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) \
-    .dtype_format(DataType.U32_Default, DataType.I32_Default, DataType.U32_Default) \
-    .dtype_format(DataType.U32_Default, DataType.I64_Default, DataType.U32_Default) \
-    .get_op_info()
-
-
-@op_info_register(broadcast_to_op_info)
-def _broadcast_to_ds_tbe():
-    """BroadcastTo TBE register"""
-    return

mindspore/ops/_op_impl/tbe/cast.py

@@ -1,55 +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.
-# ============================================================================
-
-"""Cast op"""
-from mindspore.ops.op_info_register import op_info_register, TBERegOp, DataType
-
-cast_op_info = TBERegOp("Cast") \
-    .fusion_type("OPAQUE") \
-    .async_flag(False) \
-    .binfile_name("cast.so") \
-    .compute_cost(10) \
-    .kernel_name("cast") \
-    .partial_flag(True) \
-    .attr("dst_type", "required", "int", "all") \
-    .need_check_supported(True) \
-    .input(0, "x", False, "required", "all") \
-    .output(0, "y", False, "required", "all") \
-    .op_pattern("formatAgnostic") \
-    .dtype_format(DataType.BOOL_None, DataType.F16_None) \
-    .dtype_format(DataType.BOOL_None, DataType.U8_None) \
-    .dtype_format(DataType.BOOL_None, DataType.F32_None) \
-    .dtype_format(DataType.BOOL_None, DataType.I32_None) \
-    .dtype_format(DataType.I8_None, DataType.F16_None) \
-    .dtype_format(DataType.I8_None, DataType.F32_None) \
-    .dtype_format(DataType.I8_None, DataType.I32_None) \
-    .dtype_format(DataType.U8_None, DataType.F16_None) \
-    .dtype_format(DataType.U8_None, DataType.F32_None) \
-    .dtype_format(DataType.U8_None, DataType.I32_None) \
-    .dtype_format(DataType.I32_None, DataType.BOOL_None) \
-    .dtype_format(DataType.I32_None, DataType.F16_None) \
-    .dtype_format(DataType.I32_None, DataType.F32_None) \
-    .dtype_format(DataType.F16_None, DataType.U8_None) \
-    .dtype_format(DataType.F16_None, DataType.F32_None) \
-    .dtype_format(DataType.F16_None, DataType.I32_None) \
-    .dtype_format(DataType.F32_None, DataType.F16_None) \
-    .dtype_format(DataType.F32_None, DataType.I32_None) \
-    .get_op_info()
-
-
-@op_info_register(cast_op_info)
-def _cast_tbe():
-    """Cast TBE register"""
-    return

mindspore/ops/_op_impl/tbe/cast_ds.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.
-# ============================================================================
-
-"""Cast op"""
-from mindspore.ops.op_info_register import op_info_register, TBERegOp, DataType
-
-cast_ds_op_info = TBERegOp("Cast") \
-    .fusion_type("OPAQUE") \
-    .async_flag(False) \
-    .binfile_name("cast.so") \
-    .compute_cost(10) \
-    .kernel_name("cast") \
-    .partial_flag(True) \
-    .need_check_supported(True) \
-    .attr("dst_type", "required", "int", "all") \
-    .dynamic_shape(True) \
-    .dynamic_compile_static(True) \
-    .input(0, "x", False, "required", "all") \
-    .output(0, "y", False, "required", "all") \
-    .op_pattern("formatAgnostic") \
-    .dtype_format(DataType.BOOL_None, DataType.F16_None) \
-    .dtype_format(DataType.BOOL_None, DataType.U8_None) \
-    .dtype_format(DataType.BOOL_None, DataType.F32_None) \
-    .dtype_format(DataType.BOOL_None, DataType.I32_None) \
-    .dtype_format(DataType.I8_None, DataType.F16_None) \
-    .dtype_format(DataType.I8_None, DataType.F32_None) \
-    .dtype_format(DataType.I8_None, DataType.I32_None) \
-    .dtype_format(DataType.U8_None, DataType.F16_None) \
-    .dtype_format(DataType.U8_None, DataType.F32_None) \
-    .dtype_format(DataType.U8_None, DataType.I32_None) \
-    .dtype_format(DataType.I32_None, DataType.BOOL_None) \
-    .dtype_format(DataType.I32_None, DataType.F16_None) \
-    .dtype_format(DataType.I32_None, DataType.F32_None) \
-    .dtype_format(DataType.F16_None, DataType.U8_None) \
-    .dtype_format(DataType.F16_None, DataType.F32_None) \
-    .dtype_format(DataType.F16_None, DataType.I32_None) \
-    .dtype_format(DataType.F16_None, DataType.BOOL_None) \
-    .dtype_format(DataType.F32_None, DataType.F16_None) \
-    .dtype_format(DataType.F32_None, DataType.I32_None) \
-    .get_op_info()
-
-
-@op_info_register(cast_ds_op_info)
-def _cast_ds_tbe():
-    """Cast TBE register"""
-    return

mindspore/ops/_op_impl/tbe/cdist.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.
-# ============================================================================
-
-"""Cdist op"""
-from mindspore.ops.op_info_register import op_info_register, TBERegOp, DataType
-
-cdist_op_info = TBERegOp("Cdist") \
-    .fusion_type("OPAQUE") \
-    .async_flag(False) \
-    .binfile_name("cdist.so") \
-    .compute_cost(10) \
-    .kernel_name("cdist") \
-    .partial_flag(True) \
-    .attr("p", "optional", "float", "all", "2.0") \
-    .input(0, "input_x", False, "required", "all") \
-    .input(1, "input_y", False, "required", "all") \
-    .output(0, "output", 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) \
-    .get_op_info()
-
-
-@op_info_register(cdist_op_info)
-def _cdist_tbe():
-    """Cdist TBE register"""
-    return

mindspore/ops/_op_impl/tbe/cdist_grad.py

@@ -1,42 +0,0 @@
-# Copyright 2021 Huawei Technologies Co., Ltd
-#
-# Licensed under the Apache License, Version 2.0 (the "License");
-# you may not use this file except in compliance with the License.
-# You may obtain a copy of the License at
-#
-# http://www.apache.org/licenses/LICENSE-2.0
-#
-# Unless required by applicable law or agreed to in writing, software
-# distributed under the License is distributed on an "AS IS" BASIS,
-# WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
-# See the License for the specific language governing permissions and
-# limitations under the License.
-# ============================================================================
-
-"""CdistGrad op"""
-from mindspore.ops.op_info_register import op_info_register, TBERegOp, DataType
-
-cdist_grad_op_info = TBERegOp("CdistGrad") \
-    .fusion_type("OPAQUE") \
-    .async_flag(False) \
-    .binfile_name("cdist_grad.so") \
-    .compute_cost(10) \
-    .kernel_name("cdist_grad") \
-    .partial_flag(True) \
-    .attr("p", "optional", "float", "all", "2.0") \
-    .input(0, "grad", False, "required", "all") \
-    .input(1, "input_x", False, "required", "all") \
-    .input(2, "input_y", False, "required", "all") \
-    .input(3, "cdist", False, "required", "all") \
-    .output(0, "output", False, "required", "all") \
-    .dtype_format(DataType.F16_Default, DataType.F16_Default, DataType.F16_Default, DataType.F16_Default,
-                  DataType.F16_Default) \
-    .dtype_format(DataType.F32_Default, DataType.F32_Default, DataType.F32_Default, DataType.F32_Default,
-                  DataType.F32_Default) \
-    .get_op_info()
-
-
-@op_info_register(cdist_grad_op_info)
-def _cdist_grad_tbe():
-    """CdistGrad TBE register"""
-    return

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

mindspore/ops/_op_impl/tbe/ceil_ds.py

@@ -1,38 +0,0 @@
-# Copyright 2021 Huawei Technologies Co., Ltd
-#
-# Licensed under the Apache License, Version 2.0 (the "License");
-# you may not use this file except in compliance with the License.
-# You may obtain a copy of the License at
-#
-# http://www.apache.org/licenses/LICENSE-2.0
-#
-# Unless required by applicable law or agreed to in writing, software
-# distributed under the License is distributed on an "AS IS" BASIS,
-# WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
-# See the License for the specific language governing permissions and
-# limitations under the License.
-# ============================================================================
-
-"""Ceil op"""
-from mindspore.ops.op_info_register import op_info_register, TBERegOp, DataType
-
-ceil_op_info = TBERegOp("Ceil") \
-    .fusion_type("ELEMWISE") \
-    .async_flag(False) \
-    .binfile_name("ceil.so") \
-    .compute_cost(10) \
-    .kernel_name("ceil") \
-    .partial_flag(True) \
-    .dynamic_shape(True) \
-    .input(0, "x", False, "required", "all") \
-    .output(0, "y", False, "required", "all") \
-    .op_pattern("formatAgnostic") \
-    .dtype_format(DataType.F16_None, DataType.F16_None) \
-    .dtype_format(DataType.F32_None, DataType.F32_None) \
-    .get_op_info()
-
-
-@op_info_register(ceil_op_info)
-def _ceil_ds_tbe():
-    """Ceil TBE register"""
-    return

mindspore/ops/_op_impl/tbe/celu.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.
-# ============================================================================
-
-"""Celu op"""
-from mindspore.ops.op_info_register import op_info_register, TBERegOp, DataType
-
-celu_op_info = TBERegOp("CeLU") \
-    .fusion_type("ELEMWISE") \
-    .async_flag(False) \
-    .binfile_name("celu_v2.so") \
-    .compute_cost(10) \
-    .kernel_name("celu_v2") \
-    .dynamic_compile_static(True) \
-    .dynamic_shape(True) \
-    .partial_flag(True) \
-    .attr("alpha", "optional", "float", "all", "1.0") \
-    .input(0, "x", False, "required", "all") \
-    .output(0, "y", False, "required", "all") \
-    .dtype_format(DataType.F16_Default, DataType.F16_Default) \
-    .dtype_format(DataType.F32_Default, DataType.F32_Default) \
-    .get_op_info()
-
-
-@op_info_register(celu_op_info)
-def _celu_tbe():
-    """CeLU TBE register"""
-    return

mindspore/ops/_op_impl/tbe/centralization.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.
-# ============================================================================
-
-"""Centralization op"""
-from mindspore.ops.op_info_register import op_info_register, TBERegOp, DataType
-
-centralization_op_info = TBERegOp("Centralization") \
-    .fusion_type("OPAQUE") \
-    .async_flag(False) \
-    .binfile_name("centralization.so") \
-    .compute_cost(10) \
-    .kernel_name("centralization") \
-    .partial_flag(True) \
-    .attr("axis", "required", "listInt", "all") \
-    .input(0, "x", False, "required", "all") \
-    .output(0, "y", False, "required", "all") \
-    .dtype_format(DataType.F16_Default, DataType.F16_Default) \
-    .dtype_format(DataType.F32_Default, DataType.F32_Default) \
-    .dtype_format(DataType.F16_FracZ, DataType.F16_FracZ) \
-    .dtype_format(DataType.F32_FracZ, DataType.F32_FracZ) \
-    .get_op_info()
-
-
-@op_info_register(centralization_op_info)
-def _centralization_tbe():
-    """Centralization TBE register"""
-    return

mindspore/ops/_op_impl/tbe/check_valid.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.
-# ============================================================================
-
-"""CheckValid op"""
-from mindspore.ops.op_info_register import op_info_register, TBERegOp, DataType
-
-check_valid_op_info = TBERegOp("CheckValid") \
-    .fusion_type("OPAQUE") \
-    .async_flag(False) \
-    .binfile_name("check_valid.so") \
-    .compute_cost(10) \
-    .kernel_name("check_valid") \
-    .partial_flag(True) \
-    .input(0, "bbox_tensor", False, "required", "all") \
-    .input(1, "img_tas", False, "required", "all") \
-    .output(0, "valid_tensor", False, "required", "all") \
-    .op_pattern("broadcast") \
-    .dtype_format(DataType.F16_None, DataType.F16_None, DataType.I8_None) \
-    .dtype_format(DataType.F16_None, DataType.F16_None, DataType.BOOL_None) \
-    .get_op_info()
-
-
-@op_info_register(check_valid_op_info)
-def _check_valid_tbe():
-    """CheckValid TBE register"""
-    return