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

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

mindspore/ops/_op_impl/tbe/equal_ds.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.
-# ============================================================================
-
-"""Equal op"""
-from mindspore.ops.op_info_register import op_info_register, TBERegOp, DataType
-
-equal_ds_op_info = TBERegOp("Equal") \
-    .fusion_type("ELEMWISE") \
-    .async_flag(False) \
-    .binfile_name("equal.so") \
-    .compute_cost(10) \
-    .kernel_name("equal") \
-    .partial_flag(True) \
-    .dynamic_shape(True)\
-    .input(0, "x1", False, "required", "all") \
-    .input(1, "x2", False, "required", "all") \
-    .output(0, "y", False, "required", "all") \
-    .op_pattern("broadcast") \
-    .dtype_format(DataType.I8_Default, DataType.I8_Default, DataType.BOOL_Default) \
-    .dtype_format(DataType.U8_Default, DataType.U8_Default, DataType.BOOL_Default) \
-    .dtype_format(DataType.I32_Default, DataType.I32_Default, DataType.BOOL_Default) \
-    .dtype_format(DataType.F16_Default, DataType.F16_Default, DataType.BOOL_Default) \
-    .dtype_format(DataType.F32_Default, DataType.F32_Default, DataType.BOOL_Default) \
-    .get_op_info()
-
-
-@op_info_register(equal_ds_op_info)
-def _equal_ds_tbe():
-    """Equal TBE register"""
-    return

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

mindspore/ops/_op_impl/tbe/erf_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.
-# ============================================================================
-
-"""Erf op"""
-from mindspore.ops.op_info_register import op_info_register, TBERegOp, DataType
-
-erf_op_info = TBERegOp("Erf") \
-    .fusion_type("ELEMWISE") \
-    .async_flag(False) \
-    .binfile_name("erf.so") \
-    .compute_cost(10) \
-    .kernel_name("erf") \
-    .partial_flag(True) \
-    .dynamic_shape(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(erf_op_info)
-def _erf_ds_op_tbe():
-    """Erf TBE register"""
-    return

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

mindspore/ops/_op_impl/tbe/erfc_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.
-# ============================================================================
-
-"""Erfc op"""
-from mindspore.ops.op_info_register import op_info_register, TBERegOp, DataType
-
-erfc_op_info = TBERegOp("Erfc") \
-    .fusion_type("ELEMWISE") \
-    .async_flag(False) \
-    .binfile_name("erfc.so") \
-    .compute_cost(10) \
-    .kernel_name("erfc") \
-    .partial_flag(True) \
-    .dynamic_shape(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(erfc_op_info)
-def _erfc_ds_op_tbe():
-    """Erfc TBE register"""
-    return

mindspore/ops/_op_impl/tbe/erfinv.py

@@ -1,36 +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.
-# ============================================================================
-
-"""Erfinv op"""
-from mindspore.ops.op_info_register import op_info_register, TBERegOp, DataType
-
-erfinv_op_info = TBERegOp("Erfinv") \
-    .fusion_type("ELEMWISE") \
-    .async_flag(False) \
-    .binfile_name("erfinv.so") \
-    .compute_cost(10) \
-    .kernel_name("erfinv") \
-    .partial_flag(True) \
-    .input(0, "input_x", False, "required", "all") \
-    .output(0, "output", 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(erfinv_op_info)
-def _erfinv_tbe():
-    """Erfinv TBE register"""
-    return

mindspore/ops/_op_impl/tbe/exp.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.
-# ============================================================================
-
-"""Exp op"""
-from mindspore.ops.op_info_register import op_info_register, TBERegOp, DataType
-
-exp_op_info = TBERegOp("Exp") \
-    .fusion_type("ELEMWISE") \
-    .async_flag(False) \
-    .binfile_name("exp.so") \
-    .compute_cost(10) \
-    .kernel_name("exp") \
-    .partial_flag(True) \
-    .attr("base", "optional", "float", "all", "-1.0") \
-    .attr("scale", "optional", "float", "all", "1.0") \
-    .attr("shift", "optional", "float", "all", "0.0") \
-    .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(exp_op_info)
-def _exp_tbe():
-    """Exp TBE register"""
-    return

mindspore/ops/_op_impl/tbe/exp_ds.py

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

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

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

mindspore/ops/_op_impl/tbe/expm1_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.
-# ============================================================================
-
-"""Expm1 op"""
-from mindspore.ops.op_info_register import op_info_register, TBERegOp, DataType
-
-expm1_op_info = TBERegOp("Expm1") \
-    .fusion_type("ELEMWISE") \
-    .async_flag(False) \
-    .binfile_name("expm1.so") \
-    .compute_cost(10) \
-    .kernel_name("expm1") \
-    .partial_flag(True) \
-    .dynamic_shape(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(expm1_op_info)
-def _expm1_ds_tbe():
-    """Expm1 TBE register"""
-    return

mindspore/ops/_op_impl/tbe/extract_image_patches.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.
-# ============================================================================
-
-"""ExtractImagePatches op"""
-from mindspore.ops.op_info_register import op_info_register, TBERegOp, DataType
-
-extract_image_patches_op_info = TBERegOp("ExtractImagePatches") \
-    .fusion_type("OPAQUE") \
-    .async_flag(False) \
-    .binfile_name("extract_image_patches.so") \
-    .compute_cost(10) \
-    .kernel_name("extract_image_patches") \
-    .partial_flag(True) \
-    .attr("ksizes", "required", "listInt", "all") \
-    .attr("strides", "required", "listInt", "all") \
-    .attr("rates", "required", "listInt", "all") \
-    .attr("padding", "required", "str", "all") \
-    .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(extract_image_patches_op_info)
-def _extract_image_patches_tbe():
-    """ExtractImagePatches TBE register"""
-    return

mindspore/ops/_op_impl/tbe/extract_volume_patches.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.
-# ============================================================================
-
-"""ExtractVolumePatches op"""
-from mindspore.ops.op_info_register import op_info_register, TBERegOp, DataType
-
-extract_volume_patches_op_info = TBERegOp("ExtractVolumePatches") \
-    .fusion_type("OPAQUE") \
-    .async_flag(False) \
-    .binfile_name("extract_volume_patches.so") \
-    .compute_cost(10) \
-    .kernel_name("extract_volume_patches") \
-    .partial_flag(True) \
-    .attr("kernel_size", "required", "listInt", "all") \
-    .attr("strides", "required", "listInt", "all") \
-    .attr("padding", "required", "str", "all") \
-    .input(0, "input_x", False, "required", "all") \
-    .output(0, "output_y", False, "required", "all") \
-    .dtype_format(DataType.F16_NDC1HWC0, DataType.F16_NDC1HWC0) \
-    .dtype_format(DataType.I8_NDC1HWC0, DataType.I8_NDC1HWC0) \
-    .dtype_format(DataType.U8_NDC1HWC0, DataType.U8_NDC1HWC0) \
-    .get_op_info()
-
-@op_info_register(extract_volume_patches_op_info)
-def _extract_volume_patches_tbe():
-    """ExtractVolumePatches TBE register"""
-    return

mindspore/ops/_op_impl/tbe/fake_quant_with_min_max_vars.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.
-# ============================================================================
-
-"""FakeQuantWithMinMaxVars op"""
-from mindspore.ops.op_info_register import op_info_register, TBERegOp, DataType
-
-fake_quant_with_min_max_vars_op_info = TBERegOp("FakeQuantWithMinMaxVars") \
-    .fusion_type("OPAQUE") \
-    .async_flag(False) \
-    .binfile_name("fake_quant_with_min_max_vars.so") \
-    .compute_cost(10) \
-    .kernel_name("fake_quant_with_min_max_vars") \
-    .partial_flag(True) \
-    .attr("num_bits", "optional", "int", "all") \
-    .attr("narrow_range", "optional", "bool", "all") \
-    .input(0, "x", False, "required", "all") \
-    .input(1, "min", False, "required", "all") \
-    .input(2, "max", False, "required", "all") \
-    .output(0, "y", True, "required", "all") \
-    .dtype_format(DataType.F32_Default, DataType.F32_Default, DataType.F32_Default, DataType.F32_Default) \
-    .get_op_info()
-
-
-@op_info_register(fake_quant_with_min_max_vars_op_info)
-def _fake_quant_with_min_max_vars_tbe():
-    """FakeQuantWithMinMaxVar TBE register"""
-    return

mindspore/ops/_op_impl/tbe/fake_quant_with_min_max_vars_gradient.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.
-# ============================================================================
-
-"""FakeQuantWithMinMaxVars op"""
-from mindspore.ops.op_info_register import op_info_register, TBERegOp, DataType
-
-fake_quant_with_min_max_vars_gradient_op_info = TBERegOp("FakeQuantWithMinMaxVarsGradient") \
-    .fusion_type("OPAQUE") \
-    .async_flag(False) \
-    .binfile_name("fake_quant_with_min_max_vars_gradient.so") \
-    .compute_cost(10) \
-    .kernel_name("fake_quant_with_min_max_vars_gradient") \
-    .partial_flag(True) \
-    .attr("num_bits", "optional", "int", "all") \
-    .attr("narrow_range", "optional", "bool", "all") \
-    .input(0, "gradients", False, "required", "all") \
-    .input(1, "x", False, "required", "all") \
-    .input(2, "min", False, "required", "all") \
-    .input(3, "max", False, "required", "all") \
-    .output(0, "backprops_wrt_x", True, "required", "all") \
-    .output(1, "backprops_wrt_min", True, "required", "all") \
-    .output(2, "backprops_wrt_max", True, "required", "all") \
-    .dtype_format(DataType.F32_Default, DataType.F32_Default, DataType.F32_Default, DataType.F32_Default,
-                  DataType.F32_Default, DataType.F32_Default, DataType.F32_Default) \
-    .get_op_info()
-
-
-@op_info_register(fake_quant_with_min_max_vars_gradient_op_info)
-def _fake_quant_with_min_max_vars_gradient_tbe():
-    """FakeQuantWithMinMaxVarsGradient TBE register"""
-    return