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

mindspore/ops/_op_impl/tbe/resize_bilinear_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.
-# ============================================================================
-
-"""ResizeBilinearGrad op"""
-from mindspore.ops.op_info_register import op_info_register, TBERegOp, DataType
-
-resize_bilinear_grad_op_info = TBERegOp("ResizeBilinearGrad") \
-    .fusion_type("OPAQUE") \
-    .async_flag(False) \
-    .binfile_name("resize_bilinear_v2_grad.so") \
-    .compute_cost(10) \
-    .kernel_name("resize_bilinear_v2_grad") \
-    .dynamic_compile_static(True) \
-    .dynamic_shape(True) \
-    .partial_flag(True) \
-    .need_check_supported(False) \
-    .attr("align_corners", "optional", "bool", "all", "false") \
-    .attr("half_pixel_centers", "optional", "bool", "all", "false") \
-    .input(0, "grads", False, "required", "all") \
-    .input(1, "original_image", False, "required", "all") \
-    .output(0, "y", False, "required", "all") \
-    .dtype_format(DataType.F32_5HD, DataType.F32_5HD, DataType.F32_5HD) \
-    .get_op_info()
-
-
-@op_info_register(resize_bilinear_grad_op_info)
-def _resize_bilinear_grad_tbe():
-    """ResizeBilinearGrad TBE register"""
-    return

mindspore/ops/_op_impl/tbe/resize_bilinear_v2.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.
-# ============================================================================
-
-"""ResizeBilinear op"""
-from mindspore.ops.op_info_register import op_info_register, TBERegOp, DataType
-
-resize_bilinear_v2_op_info = TBERegOp("ResizeBilinearV2") \
-    .fusion_type("OPAQUE") \
-    .async_flag(False) \
-    .binfile_name("resize_bilinear_v2.so") \
-    .compute_cost(10) \
-    .kernel_name("resize_bilinear_v2") \
-    .partial_flag(True) \
-    .need_check_supported(False) \
-    .dynamic_compile_static(True) \
-    .dynamic_shape(True) \
-    .attr("align_corners", "optional", "bool", "all", "false") \
-    .attr("half_pixel_centers", "optional", "bool", "all", "false") \
-    .input(0, "x", False, "required", "all") \
-    .input(1, "size", False, "required", "all") \
-    .output(0, "y", False, "required", "all") \
-    .dtype_format(DataType.F16_5HD, DataType.I32_Default, DataType.F32_5HD) \
-    .dtype_format(DataType.F32_5HD, DataType.I32_Default, DataType.F32_5HD) \
-    .dtype_format(DataType.F16_5HD, DataType.I32_Default, DataType.F16_5HD) \
-    .get_op_info()
-
-
-@op_info_register(resize_bilinear_v2_op_info)
-def _resize_bilinear_v2_tbe():
-    """ResizeBilinear TBE register"""
-    return

mindspore/ops/_op_impl/tbe/resize_nearest_neighbor.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.
-# ============================================================================
-
-"""ResizeNearestNeighbor op"""
-from mindspore.ops.op_info_register import op_info_register, TBERegOp, DataType
-
-resize_nearest_neighbor_op_info = TBERegOp("ResizeNearestNeighbor") \
-    .fusion_type("OPAQUE") \
-    .async_flag(False) \
-    .binfile_name("resize_nearest_neighbor_d.so") \
-    .compute_cost(10) \
-    .kernel_name("resize_nearest_neighbor_v2_d") \
-    .partial_flag(True) \
-    .need_check_supported(True) \
-    .attr("size", "required", "listInt", "all") \
-    .attr("align_corners", "optional", "bool", "all", "false") \
-    .attr("half_pixel_centers", "optional", "bool", "all", "false") \
-    .input(0, "images", False, "required", "all") \
-    .output(0, "y", True, "required", "all") \
-    .dtype_format(DataType.F16_5HD, DataType.F16_5HD) \
-    .dtype_format(DataType.F32_5HD, DataType.F32_5HD) \
-    .get_op_info()
-
-
-@op_info_register(resize_nearest_neighbor_op_info)
-def _resize_nearest_neighbor_tbe():
-    """ResizeNearestNeighbor TBE register"""
-    return

mindspore/ops/_op_impl/tbe/resize_nearest_neighbor_ds.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.
-# ============================================================================
-
-"""ResizeNearestNeighbor op"""
-from mindspore.ops.op_info_register import op_info_register, TBERegOp, DataType
-
-resize_nearest_neighbor_op_info = TBERegOp("DynamicResizeNearestNeighbor") \
-    .fusion_type("OPAQUE") \
-    .async_flag(False) \
-    .binfile_name("resize_nearest_neighbor.so") \
-    .compute_cost(10) \
-    .kernel_name("resize_nearest_neighbor_v2") \
-    .partial_flag(True) \
-    .dynamic_shape(True) \
-    .attr("align_corners", "optional", "bool", "all", "false") \
-    .attr("half_pixel_centers", "optional", "bool", "all", "false") \
-    .input(0, "x", False, "required", "all") \
-    .input(1, "size", False, "required", "all") \
-    .output(0, "y", True, "required", "all") \
-    .dtype_format(DataType.F16_5HD, DataType.I32_Default, DataType.F16_5HD) \
-    .dtype_format(DataType.F32_5HD, DataType.I32_Default, DataType.F32_5HD) \
-    .get_op_info()
-
-
-@op_info_register(resize_nearest_neighbor_op_info)
-def _resize_nearest_neighbor_ds_tbe():
-    """ResizeNearestNeighbor TBE register"""
-    return

mindspore/ops/_op_impl/tbe/resize_nearest_neighbor_grad.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.
-# ============================================================================
-
-"""ResizeNearestNeighborgrad op"""
-from mindspore.ops.op_info_register import op_info_register, TBERegOp, DataType
-
-resize_nearest_neighbor_grad_op_info = TBERegOp("ResizeNearestNeighborGrad") \
-    .fusion_type("OPAQUE") \
-    .async_flag(False) \
-    .binfile_name("resize_nearest_neighbor_grad_d.so") \
-    .compute_cost(10) \
-    .kernel_name("resize_nearest_neighbor_v2_grad_d") \
-    .partial_flag(True) \
-    .need_check_supported(True) \
-    .attr("size", "required", "listInt", "all") \
-    .attr("align_corners", "optional", "bool", "all", "false") \
-    .attr("half_pixel_centers", "optional", "bool", "all", "false") \
-    .input(0, "grads", False, "required", "all") \
-    .output(0, "y", False, "required", "all") \
-    .dtype_format(DataType.F32_5HD, DataType.F32_5HD) \
-    .get_op_info()
-
-
-@op_info_register(resize_nearest_neighbor_grad_op_info)
-def _resize_nearest_neighbor_grad_tbe():
-    """ResizeNearestNeighborGrad TBE register"""
-    return

mindspore/ops/_op_impl/tbe/resize_nearest_neighbor_grad_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.
-# ============================================================================
-
-"""ResizeNearestNeighborgrad op"""
-from mindspore.ops.op_info_register import op_info_register, TBERegOp, DataType
-
-resize_nearest_neighbor_grad_op_info = TBERegOp("ResizeNearestNeighborGrad") \
-    .fusion_type("OPAQUE") \
-    .async_flag(False) \
-    .binfile_name("resize_nearest_neighbor_grad.so") \
-    .compute_cost(10) \
-    .kernel_name("resize_nearest_neighbor_v2_grad") \
-    .partial_flag(True) \
-    .dynamic_shape(True) \
-    .attr("size", "required", "listInt", "all") \
-    .attr("align_corners", "optional", "bool", "all", "false") \
-    .attr("half_pixel_centers", "optional", "bool", "all", "false") \
-    .input(0, "grads", False, "required", "all") \
-    .input(1, "size", False, "required", "all") \
-    .output(0, "y", False, "required", "all") \
-    .dtype_format(DataType.F32_5HD, DataType.I32_Default, DataType.F32_5HD) \
-    .dtype_format(DataType.F32_5HD, DataType.I64_Default, DataType.F32_5HD) \
-    .get_op_info()
-
-
-@op_info_register(resize_nearest_neighbor_grad_op_info)
-def _resize_nearest_neighbor_grad_ds_tbe():
-    """ResizeNearestNeighborGrad TBE register"""
-    return

mindspore/ops/_op_impl/tbe/reverse_v2_d.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.
-# ============================================================================
-
-"""ReverseV2D op"""
-from mindspore.ops.op_info_register import op_info_register, TBERegOp, DataType
-
-reverse_v2_d_op_info = TBERegOp("ReverseV2") \
-    .fusion_type("ELEMWISE") \
-    .async_flag(False) \
-    .binfile_name("reverse_v2_d.so") \
-    .compute_cost(10) \
-    .kernel_name("reverse_v2_d") \
-    .partial_flag(True) \
-    .is_dynamic_format(True) \
-    .attr("axis", "required", "listInt", "all") \
-    .input(0, "x", False, "required", "all") \
-    .output(0, "y", False, "required", "all") \
-    .dtype_format(DataType.None_None, DataType.None_None) \
-    .get_op_info()
-
-
-@op_info_register(reverse_v2_d_op_info)
-def _reverse_v2_d_tbe():
-    """ReverseV2D TBE register"""
-    return

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

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

mindspore/ops/_op_impl/tbe/roi_align.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.
-# ============================================================================
-
-"""ROIAlign op"""
-from mindspore.ops.op_info_register import op_info_register, TBERegOp, DataType
-
-roi_align_op_info = TBERegOp("ROIAlign") \
-    .fusion_type("OPAQUE") \
-    .async_flag(False) \
-    .binfile_name("roi_align.so") \
-    .compute_cost(10) \
-    .kernel_name("roi_align") \
-    .partial_flag(True) \
-    .attr("spatial_scale", "required", "float", "all") \
-    .attr("pooled_height", "required", "int", "all") \
-    .attr("pooled_width", "required", "int", "all") \
-    .attr("sample_num", "optional", "int", "all", "2") \
-    .attr("roi_end_mode", "optional", "int", "0,1", "1") \
-    .input(0, "features", False, "required", "all") \
-    .input(1, "rois", False, "required", "all") \
-    .input(2, "rois_n", False, "optional", "all") \
-    .output(0, "y", False, "required", "all") \
-    .dtype_format(DataType.F16_5HD, DataType.F16_Default, DataType.I32_Default, DataType.F16_5HD) \
-    .dtype_format(DataType.F32_5HD, DataType.F32_Default, DataType.I32_Default, DataType.F32_5HD) \
-    .get_op_info()
-
-
-@op_info_register(roi_align_op_info)
-def _roi_align_tbe():
-    """ROIAlign TBE register"""
-    return

mindspore/ops/_op_impl/tbe/roi_align_ds.py

@@ -1,44 +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.
-# ============================================================================
-
-"""ROIAlign op"""
-from mindspore.ops.op_info_register import op_info_register, TBERegOp, DataType
-
-roi_align_op_info = TBERegOp("ROIAlign") \
-    .fusion_type("OPAQUE") \
-    .async_flag(False) \
-    .binfile_name("roi_align.so") \
-    .compute_cost(10) \
-    .kernel_name("roi_align") \
-    .partial_flag(True) \
-    .dynamic_shape(True) \
-    .attr("spatial_scale", "required", "float", "all") \
-    .attr("pooled_height", "required", "int", "all") \
-    .attr("pooled_width", "required", "int", "all") \
-    .attr("sample_num", "optional", "int", "all", "2") \
-    .attr("roi_end_mode", "optional", "int", "0,1", "1") \
-    .input(0, "features", False, "required", "all") \
-    .input(1, "rois", False, "required", "all") \
-    .input(2, "rois_n", False, "optional", "all") \
-    .output(0, "y", False, "required", "all") \
-    .dtype_format(DataType.F16_5HD, DataType.F16_Default, DataType.I32_Default, DataType.F16_5HD) \
-    .dtype_format(DataType.F32_5HD, DataType.F32_Default, DataType.I32_Default, DataType.F32_5HD) \
-    .get_op_info()
-
-
-@op_info_register(roi_align_op_info)
-def _roi_align_ds_tbe():
-    """ROIAlign TBE register"""
-    return

mindspore/ops/_op_impl/tbe/roi_align_grad.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.
-# ============================================================================
-
-"""ROIAlignGrad op"""
-from mindspore.ops.op_info_register import op_info_register, TBERegOp, DataType
-
-roi_align_grad_op_info = TBERegOp("ROIAlignGrad") \
-    .fusion_type("OPAQUE") \
-    .async_flag(False) \
-    .binfile_name("roi_align_grad.so") \
-    .compute_cost(10) \
-    .kernel_name("roi_align_grad") \
-    .partial_flag(True) \
-    .attr("xdiff_shape", "required", "listInt", "all") \
-    .attr("pooled_width", "required", "int", "all") \
-    .attr("pooled_height", "required", "int", "all") \
-    .attr("spatial_scale", "required", "float", "all") \
-    .attr("sample_num", "optional", "int", "all") \
-    .attr("roi_end_mode", "optional", "int", "0,1,2", "1") \
-    .input(0, "ydiff", False, "required", "all") \
-    .input(1, "rois", False, "required", "all") \
-    .input(2, "rois_n", False, "optional", "all") \
-    .output(0, "xdiff", False, "required", "all") \
-    .dtype_format(DataType.F32_5HD, DataType.F32_Default, DataType.I32_Default, DataType.F32_5HD) \
-    .get_op_info()
-
-
-@op_info_register(roi_align_grad_op_info)
-def _roi_align_grad_tbe():
-    """ROIAlignGrad TBE register"""
-    return

mindspore/ops/_op_impl/tbe/roi_align_grad_ds.py

@@ -1,44 +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.
-# ============================================================================
-
-"""ROIAlignGrad op"""
-from mindspore.ops.op_info_register import op_info_register, TBERegOp, DataType
-
-roi_align_grad_op_info = TBERegOp("ROIAlignGrad") \
-    .fusion_type("OPAQUE") \
-    .async_flag(False) \
-    .binfile_name("roi_align_grad.so") \
-    .compute_cost(10) \
-    .kernel_name("roi_align_grad") \
-    .partial_flag(True) \
-    .dynamic_shape(True) \
-    .attr("xdiff_shape", "required", "listInt", "all") \
-    .attr("pooled_width", "required", "int", "all") \
-    .attr("pooled_height", "required", "int", "all") \
-    .attr("spatial_scale", "required", "float", "all") \
-    .attr("sample_num", "optional", "int", "all") \
-    .attr("roi_end_mode", "optional", "int", "0,1,2", "1") \
-    .input(0, "ydiff", False, "required", "all") \
-    .input(1, "rois", False, "required", "all") \
-    .input(2, "rois_n", False, "optional", "all") \
-    .output(0, "xdiff", False, "required", "all") \
-    .dtype_format(DataType.F32_5HD, DataType.F32_Default, DataType.I32_Default, DataType.F32_5HD) \
-    .get_op_info()
-
-
-@op_info_register(roi_align_grad_op_info)
-def _roi_align_grad_ds_tbe():
-    """ROIAlignGrad TBE register"""
-    return

mindspore/ops/_op_impl/tbe/roll.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.
-# ============================================================================
-
-"""Roll op"""
-from mindspore.ops.op_info_register import op_info_register, TBERegOp, DataType
-
-roll_op_info = TBERegOp("Roll") \
-    .fusion_type("OPAQUE") \
-    .async_flag(False) \
-    .binfile_name("roll.so") \
-    .compute_cost(10) \
-    .kernel_name("roll") \
-    .partial_flag(True) \
-    .attr("shift", "required", "listInt", "all") \
-    .attr("axis", "optional", "listInt", "all") \
-    .input(0, "input_x", False, "required", "all") \
-    .output(0, "output", False, "required", "all") \
-    .dtype_format(DataType.F32_Default, DataType.F32_Default) \
-    .dtype_format(DataType.F16_Default, DataType.F16_Default) \
-    .dtype_format(DataType.I32_Default, DataType.I32_Default) \
-    .dtype_format(DataType.U32_Default, DataType.U32_Default) \
-    .dtype_format(DataType.I8_Default, DataType.I8_Default) \
-    .dtype_format(DataType.U8_Default, DataType.U8_Default) \
-    .get_op_info()
-
-
-@op_info_register(roll_op_info)
-def _roll_tbe():
-    """Roll TBE register"""
-    return

mindspore/ops/_op_impl/tbe/round.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.
-# ============================================================================
-
-"""Round op"""
-from mindspore.ops.op_info_register import op_info_register, TBERegOp, DataType
-
-round_op_info = TBERegOp("Round") \
-    .fusion_type("OPAQUE") \
-    .async_flag(False) \
-    .binfile_name("round.so") \
-    .compute_cost(10) \
-    .kernel_name("round") \
-    .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) \
-    .dtype_format(DataType.I32_None, DataType.I32_None) \
-    .get_op_info()
-
-
-@op_info_register(round_op_info)
-def _round_tbe():
-    """Round TBE register"""
-    return

mindspore/ops/_op_impl/tbe/round_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.
-# ============================================================================
-
-"""Round op"""
-from mindspore.ops.op_info_register import op_info_register, TBERegOp, DataType
-
-round_op_info = TBERegOp("Round") \
-    .fusion_type("OPAQUE") \
-    .async_flag(False) \
-    .binfile_name("round.so") \
-    .compute_cost(10) \
-    .kernel_name("round") \
-    .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) \
-    .dtype_format(DataType.I32_None, DataType.I32_None) \
-    .get_op_info()
-
-
-@op_info_register(round_op_info)
-def _round_ds_tbe():
-    """Round TBE register"""
-    return