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

mindspore/ops/_op_impl/tbe/hshrink_grad.py

@@ -1,37 +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.
-# ============================================================================
-"""HShrinkGrad op"""
-from mindspore.ops.op_info_register import op_info_register, TBERegOp, DataType
-
-hshrink_grad_op_info = TBERegOp("HShrinkGrad") \
-    .fusion_type("OPAQUE") \
-    .async_flag(False) \
-    .binfile_name("hard_shrink_grad.so") \
-    .compute_cost(10) \
-    .kernel_name("hard_shrink_grad") \
-    .partial_flag(True) \
-    .attr("lambd", "optional", "float", "all", "0.5") \
-    .input(0, "gradients", False, "required", "all") \
-    .input(1, "features", False, "required", "all") \
-    .output(0, "backprops", 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(hshrink_grad_op_info)
-def _hshrink_grad_tbe():
-    """HShrinkGrad TBE register"""
-    return

mindspore/ops/_op_impl/tbe/hsigmoid.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.
-# ============================================================================
-
-"""HSigmoid op"""
-from mindspore.ops.op_info_register import op_info_register, TBERegOp, DataType
-
-hsigmoid_op_info = TBERegOp("HSigmoid") \
-    .fusion_type("ELEMWISE") \
-    .async_flag(False) \
-    .binfile_name("hard_sigmoid.so") \
-    .compute_cost(10) \
-    .kernel_name("hard_sigmoid") \
-    .partial_flag(True) \
-    .attr("alpha", "optional", "float", "all", "0.16666666") \
-    .attr("beta", "optional", "float", "all", "0.5") \
-    .input(0, "input_x", False, "required", "all") \
-    .output(0, "output_y", False, "required", "all") \
-    .dtype_format(DataType.F16_Default, DataType.F16_Default) \
-    .dtype_format(DataType.F16_5HD, DataType.F16_5HD) \
-    .dtype_format(DataType.F16_NHWC, DataType.F16_NHWC) \
-    .dtype_format(DataType.F32_Default, DataType.F32_Default) \
-    .dtype_format(DataType.F32_5HD, DataType.F32_5HD) \
-    .dtype_format(DataType.F32_NHWC, DataType.F32_NHWC) \
-    .dtype_format(DataType.I32_Default, DataType.I32_Default) \
-    .dtype_format(DataType.I32_5HD, DataType.I32_5HD) \
-    .dtype_format(DataType.I32_NHWC, DataType.I32_NHWC) \
-    .get_op_info()
-
-
-@op_info_register(hsigmoid_op_info)
-def _hsigmoid_tbe():
-    """HSigmoid TBE register"""
-    return

mindspore/ops/_op_impl/tbe/hsigmoid_grad.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.
-# ============================================================================
-
-"""HSigmoidGrad op"""
-from mindspore.ops.op_info_register import op_info_register, TBERegOp, DataType
-
-hsigmoid_grad_op_info = TBERegOp("HSigmoidGrad") \
-    .fusion_type("OPAQUE") \
-    .async_flag(False) \
-    .binfile_name("hard_sigmoid_grad.so") \
-    .compute_cost(10) \
-    .kernel_name("hard_sigmoid_grad") \
-    .partial_flag(True) \
-    .attr("alpha", "optional", "float", "all", "0.16666666") \
-    .attr("beta", "optional", "float", "all", "0.5") \
-    .input(0, "grads", False, "required", "all") \
-    .input(1, "input_x", 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(hsigmoid_grad_op_info)
-def _hsigmoid_grad_tbe():
-    """HSigmoidGrad TBE register"""
-    return

mindspore/ops/_op_impl/tbe/ifmr.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.
-# ============================================================================
-
-"""IFMR op"""
-from mindspore.ops.op_info_register import op_info_register, TBERegOp, DataType
-
-ifmr_op_info = TBERegOp("IFMR") \
-    .fusion_type("OPAQUE") \
-    .async_flag(False) \
-    .binfile_name("ifmr.so") \
-    .compute_cost(10) \
-    .kernel_name("ifmr") \
-    .partial_flag(True) \
-    .attr("min_percentile", "required", "float", "all") \
-    .attr("max_percentile", "required", "float", "all") \
-    .attr("search_range", "required", "listFloat", "all") \
-    .attr("search_step", "required", "float", "all") \
-    .attr("with_offset", "required", "bool", "all") \
-    .input(0, "data", False, "required", "all") \
-    .input(1, "data_min", False, "required", "all") \
-    .input(2, "data_max", False, "required", "all") \
-    .input(3, "cumsum", False, "required", "all") \
-    .output(0, "scale", False, "required", "all") \
-    .output(1, "offset", False, "required", "all") \
-    .dtype_format(DataType.F16_Default, DataType.F16_Default, DataType.F16_Default, DataType.I32_Default,
-                  DataType.F16_Default, DataType.F16_Default) \
-    .dtype_format(DataType.F32_Default, DataType.F32_Default, DataType.F32_Default, DataType.I32_Default,
-                  DataType.F32_Default, DataType.F32_Default) \
-    .get_op_info()
-
-
-@op_info_register(ifmr_op_info)
-def _ifmr_tbe():
-    """IFMR TBE register"""
-    return

mindspore/ops/_op_impl/tbe/ifmr_ds.py

@@ -1,48 +0,0 @@
-# Copyright 2022 Huawei Technologies Co., Ltd
-#
-# Licensed under the Apache License, Version 2.0 (the "License");
-# you may not use this file except in compliance with the License.
-# You may obtain a copy of the License at
-#
-# http://www.apache.org/licenses/LICENSE-2.0
-#
-# Unless required by applicable law or agreed to in writing, software
-# distributed under the License is distributed on an "AS IS" BASIS,
-# WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
-# See the License for the specific language governing permissions and
-# limitations under the License.
-# ============================================================================
-
-"""IFMR op"""
-from mindspore.ops.op_info_register import op_info_register, TBERegOp, DataType
-
-ifmr_ds_op_info = TBERegOp("IFMR") \
-    .fusion_type("OPAQUE") \
-    .async_flag(False) \
-    .binfile_name("ifmr.so") \
-    .compute_cost(10) \
-    .kernel_name("ifmr") \
-    .partial_flag(True) \
-    .dynamic_shape(True) \
-    .attr("min_percentile", "required", "float", "all") \
-    .attr("max_percentile", "required", "float", "all") \
-    .attr("search_range", "required", "listFloat", "all") \
-    .attr("search_step", "required", "float", "all") \
-    .attr("with_offset", "required", "bool", "all") \
-    .input(0, "data", False, "required", "all") \
-    .input(1, "data_min", False, "required", "all") \
-    .input(2, "data_max", False, "required", "all") \
-    .input(3, "cumsum", False, "required", "all") \
-    .output(0, "scale", False, "required", "all") \
-    .output(1, "offset", False, "required", "all") \
-    .dtype_format(DataType.F16_Default, DataType.F16_Default, DataType.F16_Default, DataType.I32_Default,
-                  DataType.F16_Default, DataType.F16_Default) \
-    .dtype_format(DataType.F32_Default, DataType.F32_Default, DataType.F32_Default, DataType.I32_Default,
-                  DataType.F32_Default, DataType.F32_Default) \
-    .get_op_info()
-
-
-@op_info_register(ifmr_ds_op_info)
-def _ifmr_ds_tbe():
-    """IFMR TBE register"""
-    return

mindspore/ops/_op_impl/tbe/im2col.py

@@ -1,42 +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.
-# ============================================================================
-
-"""Im2Col op"""
-from mindspore.ops.op_info_register import op_info_register, TBERegOp, DataType
-
-im2col_op_info = TBERegOp("Im2Col") \
-    .fusion_type("OPAQUE") \
-    .async_flag(False) \
-    .binfile_name("im2col.so") \
-    .compute_cost(10) \
-    .kernel_name("im2col") \
-    .partial_flag(True) \
-    .attr("ksizes", "required", "listInt", "all", "[]") \
-    .attr("strides", "optional", "listInt", "all", "[1]") \
-    .attr("dilations", "optional", "listInt", "all", "[1]") \
-    .attr("padding_mode", "optional", "str", "all", "CALCULATED") \
-    .attr("pads", "optional", "listInt", "all", "[0]") \
-    .input(0, "x", False, "required", "all") \
-    .output(0, "y", False, "required", "all") \
-    .dtype_format(DataType.F16_5HD, DataType.F16_NHWC) \
-    .dtype_format(DataType.I8_5HD, DataType.I8_NHWC) \
-    .dtype_format(DataType.U8_5HD, DataType.U8_NHWC) \
-    .get_op_info()
-
-
-@op_info_register(im2col_op_info)
-def _im2col_tbe():
-    """Im2Col TBE register"""
-    return

mindspore/ops/_op_impl/tbe/in_top_k.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.
-# ============================================================================
-
-"""InTopK op"""
-from mindspore.ops.op_info_register import op_info_register, TBERegOp, DataType
-
-in_top_k_op_info = TBERegOp("InTopK") \
-    .fusion_type("OPAQUE") \
-    .async_flag(False) \
-    .binfile_name("in_top_k.so") \
-    .compute_cost(10) \
-    .kernel_name("in_top_k") \
-    .partial_flag(True) \
-    .attr("k", "required", "int", "all") \
-    .input(0, "x1", False, "required", "all") \
-    .input(1, "x2", False, "required", "all") \
-    .output(0, "y", False, "required", "all") \
-    .dtype_format(DataType.F32_Default, DataType.I32_Default, DataType.BOOL_Default) \
-    .get_op_info()
-
-
-@op_info_register(in_top_k_op_info)
-def _in_top_k_tbe():
-    """InTopK TBE register"""
-    return

mindspore/ops/_op_impl/tbe/inplace_add.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.
-# ============================================================================
-
-"""InplaceAdd op"""
-from mindspore.ops.op_info_register import op_info_register, TBERegOp, DataType
-
-inplace_add_op_info = TBERegOp("InplaceAdd") \
-    .fusion_type("ELEMWISE") \
-    .async_flag(False) \
-    .binfile_name("inplace_add_d.so") \
-    .compute_cost(10) \
-    .kernel_name("inplace_add_d") \
-    .partial_flag(True) \
-    .attr("indices", "required", "listInt", "all") \
-    .input(0, "x", False, "required", "all") \
-    .input(1, "v", False, "required", "all") \
-    .output(0, "y", False, "required", "all") \
-    .dtype_format(DataType.I32_Default, DataType.I32_Default, DataType.I32_Default) \
-    .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(inplace_add_op_info)
-def _inplace_add_tbe():
-    """InplaceAdd TBE register"""
-    return

mindspore/ops/_op_impl/tbe/inplace_index_add.py

@@ -1,46 +0,0 @@
-# Copyright 2021-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.
-# ============================================================================
-
-"""IndexAdd op"""
-from mindspore.ops.op_info_register import op_info_register, TBERegOp, DataType
-
-inplace_index_add_op_info = TBERegOp("InplaceIndexAdd") \
-    .fusion_type("OPAQUE") \
-    .async_flag(False) \
-    .binfile_name("inplace_index_add.so") \
-    .compute_cost(10) \
-    .kernel_name("inplace_index_add") \
-    .partial_flag(True) \
-    .dynamic_compile_static(True) \
-    .dynamic_shape(True) \
-    .attr("axis", "required", "int", "all") \
-    .input(0, "input_x", False, "required", "all") \
-    .input(1, "indices", False, "required", "all") \
-    .input(2, "input_y", False, "required", "all") \
-    .input(3, "alpha", False, "optional", "all") \
-    .output(0, "input_x", False, "required", "all") \
-    .dtype_format(DataType.F16_Default, DataType.I32_Default, DataType.F16_Default, DataType.F16_Default,
-                  DataType.F16_Default) \
-    .dtype_format(DataType.F32_Default, DataType.I32_Default, DataType.F32_Default, DataType.F32_Default,
-                  DataType.F32_Default) \
-    .dtype_format(DataType.I32_Default, DataType.I32_Default, DataType.I32_Default, DataType.I32_Default,
-                  DataType.I32_Default) \
-    .get_op_info()
-
-
-@op_info_register(inplace_index_add_op_info)
-def _inplace_index_add_tbe():
-    """IndexAdd TBE register"""
-    return

mindspore/ops/_op_impl/tbe/inplace_sub.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.
-# ============================================================================
-
-"""InplaceSub op"""
-from mindspore.ops.op_info_register import op_info_register, TBERegOp, DataType
-
-inplace_sub_op_info = TBERegOp("InplaceSub") \
-    .fusion_type("ELEMWISE") \
-    .async_flag(False) \
-    .binfile_name("inplace_sub_d.so") \
-    .compute_cost(10) \
-    .kernel_name("inplace_sub_d") \
-    .partial_flag(True) \
-    .attr("indices", "required", "listInt", "all") \
-    .input(0, "x", False, "required", "all") \
-    .input(1, "v", False, "required", "all") \
-    .output(0, "y", False, "required", "all") \
-    .dtype_format(DataType.I32_Default, DataType.I32_Default, DataType.I32_Default) \
-    .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(inplace_sub_op_info)
-def _inplace_sub_tbe():
-    """InplaceSub TBE register"""
-    return

mindspore/ops/_op_impl/tbe/inplace_update.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.
-# ============================================================================
-
-"""InplaceUpdate op"""
-from mindspore.ops.op_info_register import op_info_register, TBERegOp, DataType
-
-inplace_update_op_info = TBERegOp("InplaceUpdate") \
-    .fusion_type("INPLACE") \
-    .async_flag(False) \
-    .binfile_name("inplace_update_d.so") \
-    .compute_cost(10) \
-    .kernel_name("inplace_update_d") \
-    .partial_flag(True) \
-    .attr("indices", "required", "listInt", "all") \
-    .input(0, "x", False, "required", "all") \
-    .input(1, "v", False, "required", "all") \
-    .output(0, "y", False, "required", "all") \
-    .dtype_format(DataType.F16_Default, DataType.F16_Default, DataType.F16_Default) \
-    .dtype_format(DataType.F32_Default, DataType.F32_Default, DataType.F32_Default) \
-    .dtype_format(DataType.I32_Default, DataType.I32_Default, DataType.I32_Default) \
-    .get_op_info()
-
-
-@op_info_register(inplace_update_op_info)
-def _inplace_update_tbe():
-    """InplaceUpdate TBE register"""
-    return

mindspore/ops/_op_impl/tbe/inplace_update_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.
-# ============================================================================
-
-"""InplaceUpdate op"""
-from mindspore.ops.op_info_register import op_info_register, TBERegOp, DataType
-
-inplace_update_v2_op_info = TBERegOp("InplaceUpdateV2") \
-    .fusion_type("INPLACE") \
-    .async_flag(False) \
-    .binfile_name("inplace_update.so") \
-    .compute_cost(10) \
-    .kernel_name("inplace_update") \
-    .partial_flag(True) \
-    .dynamic_shape(True) \
-    .input(0, "x", False, "required", "all") \
-    .input(1, "indices", False, "required", "all") \
-    .input(2, "v", False, "required", "all") \
-    .output(0, "y", False, "required", "all") \
-    .dtype_format(DataType.F16_Default, DataType.I32_Default, DataType.F16_Default, DataType.F16_Default) \
-    .dtype_format(DataType.F32_Default, DataType.I32_Default, DataType.F32_Default, DataType.F32_Default) \
-    .dtype_format(DataType.I32_Default, DataType.I32_Default, DataType.I32_Default, DataType.I32_Default) \
-    .get_op_info()
-
-
-@op_info_register(inplace_update_v2_op_info)
-def _inplace_update_v2_ds_tbe():
-    """InplaceUpdate TBE register"""
-    return

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

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

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