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

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

mindspore/ops/_op_impl/tbe/fake_quant_with_min_max_vars_per_channel.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.
-# ============================================================================
-
-"""FakeQuantWithMinMaxVarsPerChannel op"""
-from mindspore.ops.op_info_register import op_info_register, TBERegOp, DataType
-
-fake_quant_with_min_max_vars_per_channel_op_info = TBERegOp("FakeQuantWithMinMaxVarsPerChannel") \
-    .fusion_type("OPAQUE") \
-    .async_flag(False) \
-    .binfile_name("fake_quant_with_min_max_vars_per_channel.so") \
-    .compute_cost(10) \
-    .kernel_name("fake_quant_with_min_max_vars_per_channel") \
-    .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_per_channel_op_info)
-def _fake_quant_with_min_max_vars_per_channel_tbe():
-    """FakeQuantWithMinMaxVarsPerChannel TBE register"""
-    return

mindspore/ops/_op_impl/tbe/fake_quant_with_min_max_vars_per_channel_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.
-# ============================================================================
-
-"""FakeQuantWithMinMaxVarsPerChannelGradient op"""
-from mindspore.ops.op_info_register import op_info_register, TBERegOp, DataType
-
-fake_quant_with_min_max_vars_per_channel_gradient_op_info = TBERegOp("FakeQuantWithMinMaxVarsPerChannelGradient") \
-    .fusion_type("OPAQUE") \
-    .async_flag(False) \
-    .binfile_name("fake_quant_with_min_max_vars_per_channel_gradient.so") \
-    .compute_cost(10) \
-    .kernel_name("fake_quant_with_min_max_vars_per_channel_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_per_channel_gradient_op_info)
-def _fake_quant_with_min_max_vars_per_channel_gradient_tbe():
-    """FakeQuantWithMinMaxVarsPerChannelGradient TBE register"""
-    return

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

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

mindspore/ops/_op_impl/tbe/fast_gelu_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.
-# ============================================================================
-
-"""FastGeLUGrad op"""
-from mindspore.ops.op_info_register import op_info_register, TBERegOp, DataType
-
-fast_gelu_grad_op_info = TBERegOp("FastGeLUGrad") \
-    .fusion_type("ELEMWISE") \
-    .async_flag(False) \
-    .binfile_name("fast_gelu_grad.so") \
-    .compute_cost(10) \
-    .kernel_name("fast_gelu_grad") \
-    .partial_flag(True) \
-    .input(0, "dy", False, "required", "all") \
-    .input(1, "x", False, "required", "all") \
-    .output(0, "z", True, "required", "all") \
-    .dtype_format(DataType.F16_Default, DataType.F16_Default, DataType.F16_Default) \
-    .dtype_format(DataType.F16_5HD, DataType.F16_5HD, DataType.F16_5HD) \
-    .dtype_format(DataType.F16_FracNZ, DataType.F16_FracNZ, DataType.F16_FracNZ) \
-    .dtype_format(DataType.F32_Default, DataType.F32_Default, DataType.F32_Default) \
-    .dtype_format(DataType.F32_5HD, DataType.F32_5HD, DataType.F32_5HD) \
-    .dtype_format(DataType.F32_FracNZ, DataType.F32_FracNZ, DataType.F32_FracNZ) \
-    .get_op_info()
-
-
-@op_info_register(fast_gelu_grad_op_info)
-def _fast_gelu_grad_tbe():
-    """FastGeLUGrad TBE register"""
-    return

mindspore/ops/_op_impl/tbe/fast_gelu_grad_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.
-# ============================================================================
-
-"""FastGeLUGrad op"""
-from mindspore.ops.op_info_register import op_info_register, TBERegOp, DataType
-
-fast_gelu_grad_op_info = TBERegOp("FastGeLUGrad") \
-    .fusion_type("ELEMWISE") \
-    .async_flag(False) \
-    .binfile_name("fast_gelu_grad.so") \
-    .compute_cost(10) \
-    .kernel_name("fast_gelu_grad") \
-    .partial_flag(True) \
-    .dynamic_shape(True) \
-    .input(0, "dy", False, "required", "all") \
-    .input(1, "x", False, "required", "all") \
-    .output(0, "z", True, "required", "all") \
-    .dtype_format(DataType.F16_Default, DataType.F16_Default, DataType.F16_Default) \
-    .dtype_format(DataType.F16_5HD, DataType.F16_5HD, DataType.F16_5HD) \
-    .dtype_format(DataType.F16_FracNZ, DataType.F16_FracNZ, DataType.F16_FracNZ) \
-    .dtype_format(DataType.F32_Default, DataType.F32_Default, DataType.F32_Default) \
-    .dtype_format(DataType.F32_5HD, DataType.F32_5HD, DataType.F32_5HD) \
-    .dtype_format(DataType.F32_FracNZ, DataType.F32_FracNZ, DataType.F32_FracNZ) \
-    .get_op_info()
-
-
-@op_info_register(fast_gelu_grad_op_info)
-def _fast_gelu_grad_ds_tbe():
-    """FastGeLUGrad TBE register"""
-    return

mindspore/ops/_op_impl/tbe/fill.py

@@ -1,56 +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.
-# ============================================================================
-
-"""FillD op"""
-from mindspore.ops.op_info_register import op_info_register, TBERegOp, DataType
-
-fill_d_op_info = TBERegOp("Fill") \
-    .fusion_type("ELEMWISE") \
-    .async_flag(False) \
-    .binfile_name("fill_d.so") \
-    .compute_cost(10) \
-    .kernel_name("fill_d") \
-    .partial_flag(True) \
-    .need_check_supported(True) \
-    .attr("dims", "required", "listInt", "all") \
-    .input(0, "value", False, "required", "all") \
-    .output(0, "y", False, "required", "all") \
-    .dtype_format(DataType.F16_5HD, DataType.F16_5HD) \
-    .dtype_format(DataType.F16_FracZ, DataType.F16_FracZ) \
-    .dtype_format(DataType.F16_C1HWNCoC0, DataType.F16_C1HWNCoC0) \
-    .dtype_format(DataType.F16_Default, DataType.F16_Default) \
-    .dtype_format(DataType.F32_5HD, DataType.F32_5HD) \
-    .dtype_format(DataType.F32_FracZ, DataType.F32_FracZ) \
-    .dtype_format(DataType.F32_C1HWNCoC0, DataType.F32_C1HWNCoC0) \
-    .dtype_format(DataType.F32_Default, DataType.F32_Default) \
-    .dtype_format(DataType.I32_5HD, DataType.I32_5HD) \
-    .dtype_format(DataType.I32_FracZ, DataType.I32_FracZ) \
-    .dtype_format(DataType.I32_C1HWNCoC0, DataType.I32_C1HWNCoC0) \
-    .dtype_format(DataType.I32_Default, DataType.I32_Default) \
-    .dtype_format(DataType.I8_5HD, DataType.I8_5HD) \
-    .dtype_format(DataType.I8_FracZ, DataType.I8_FracZ) \
-    .dtype_format(DataType.I8_C1HWNCoC0, DataType.I8_C1HWNCoC0) \
-    .dtype_format(DataType.I8_Default, DataType.I8_Default) \
-    .dtype_format(DataType.U8_5HD, DataType.U8_5HD) \
-    .dtype_format(DataType.U8_FracZ, DataType.U8_FracZ) \
-    .dtype_format(DataType.U8_C1HWNCoC0, DataType.U8_C1HWNCoC0) \
-    .dtype_format(DataType.U8_Default, DataType.U8_Default) \
-    .get_op_info()
-
-
-@op_info_register(fill_d_op_info)
-def _fill_op_tbe():
-    """FillD TBE register"""
-    return

mindspore/ops/_op_impl/tbe/fill_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.
-# ============================================================================
-
-"""Fill op"""
-from mindspore.ops.op_info_register import op_info_register, TBERegOp, DataType
-
-fill_ds_op_info = TBERegOp("Fill") \
-    .fusion_type("ELEMWISE") \
-    .async_flag(False) \
-    .binfile_name("fill.so") \
-    .compute_cost(10) \
-    .kernel_name("fill") \
-    .partial_flag(True) \
-    .dynamic_shape(True) \
-    .input(0, "dims", False, "required", "all") \
-    .input(1, "value", False, "required", "all") \
-    .output(0, "y", False, "required", "all") \
-    .dtype_format(DataType.I32_Default, DataType.I32_Default, DataType.I32_Default) \
-    .dtype_format(DataType.I32_Default, DataType.F32_Default, DataType.F32_Default) \
-    .dtype_format(DataType.I32_Default, DataType.F16_Default, DataType.I16_Default) \
-    .dtype_format(DataType.I64_Default, DataType.I32_Default, DataType.I32_Default) \
-    .dtype_format(DataType.I64_Default, DataType.F32_Default, DataType.F32_Default) \
-    .dtype_format(DataType.I64_Default, DataType.F16_Default, DataType.F16_Default) \
-    .get_op_info()
-
-
-@op_info_register(fill_ds_op_info)
-def _fill_ds_op_tbe():
-    """Fill TBE register"""
-    return

mindspore/ops/_op_impl/tbe/flatten.py

@@ -1,48 +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.
-# ============================================================================
-
-"""Flatten op"""
-from mindspore.ops.op_info_register import op_info_register, TBERegOp, DataType
-
-flatten_op_info = TBERegOp("Flatten") \
-    .fusion_type("OPAQUE") \
-    .async_flag(False) \
-    .binfile_name("flatten.so") \
-    .compute_cost(10) \
-    .kernel_name("flatten") \
-    .partial_flag(True) \
-    .dynamic_compile_static(True) \
-    .dynamic_shape(True) \
-    .attr("axis", "optional", "int", "all", "1") \
-    .input(0, "x", False, "required", "all") \
-    .output(0, "y", False, "required", "all") \
-    .dtype_format(DataType.BOOL_Default, DataType.BOOL_Default) \
-    .dtype_format(DataType.I8_Default, DataType.I8_Default) \
-    .dtype_format(DataType.U8_Default, DataType.U8_Default) \
-    .dtype_format(DataType.I16_Default, DataType.I16_Default) \
-    .dtype_format(DataType.U16_Default, DataType.U16_Default) \
-    .dtype_format(DataType.I32_Default, DataType.I32_Default) \
-    .dtype_format(DataType.U32_Default, DataType.U32_Default) \
-    .dtype_format(DataType.I64_Default, DataType.I64_Default) \
-    .dtype_format(DataType.U64_Default, DataType.U64_Default) \
-    .dtype_format(DataType.F16_Default, DataType.F16_Default) \
-    .dtype_format(DataType.F32_Default, DataType.F32_Default) \
-    .get_op_info()
-
-
-@op_info_register(flatten_op_info)
-def _flatten_tbe():
-    """Flatten TBE register"""
-    return