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

mindspore/ops/_op_impl/tbe/logical_and_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.
-# ============================================================================
-
-"""LogicalAnd op"""
-from mindspore.ops.op_info_register import op_info_register, TBERegOp, DataType
-
-logical_and_op_info = TBERegOp("LogicalAnd") \
-    .fusion_type("ELEMWISE") \
-    .async_flag(False) \
-    .binfile_name("logical_and.so") \
-    .compute_cost(10) \
-    .kernel_name("logical_and") \
-    .dynamic_shape(True) \
-    .partial_flag(True) \
-    .input(0, "x1", False, "required", "all") \
-    .input(1, "x2", False, "required", "all") \
-    .output(0, "y", True, "required", "all") \
-    .op_pattern("broadcast") \
-    .dtype_format(DataType.BOOL_None, DataType.BOOL_None, DataType.BOOL_None) \
-    .get_op_info()
-
-
-@op_info_register(logical_and_op_info)
-def _logical_and_ds_tbe():
-    """LogicalAnd TBE register"""
-    return

mindspore/ops/_op_impl/tbe/logical_not.py

@@ -1,36 +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.
-# ============================================================================
-
-"""LogicalNot op"""
-from mindspore.ops.op_info_register import op_info_register, TBERegOp, DataType
-
-logical_not_op_info = TBERegOp("LogicalNot") \
-    .fusion_type("ELEMWISE") \
-    .async_flag(False) \
-    .binfile_name("logical_not.so") \
-    .compute_cost(10) \
-    .kernel_name("logical_not") \
-    .partial_flag(True) \
-    .input(0, "x", False, "required", "all") \
-    .output(0, "y", True, "required", "all") \
-    .op_pattern("formatAgnostic") \
-    .dtype_format(DataType.BOOL_None, DataType.BOOL_None) \
-    .get_op_info()
-
-
-@op_info_register(logical_not_op_info)
-def _logical_not_tbe():
-    """LogicalNot TBE register"""
-    return

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

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

mindspore/ops/_op_impl/tbe/logical_or_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.
-# ============================================================================
-
-"""LogicalOr op"""
-from mindspore.ops.op_info_register import op_info_register, TBERegOp, DataType
-
-logical_or_op_info = TBERegOp("LogicalOr") \
-    .fusion_type("ELEMWISE") \
-    .async_flag(False) \
-    .binfile_name("logical_or.so") \
-    .compute_cost(10) \
-    .kernel_name("logical_or") \
-    .dynamic_shape(True) \
-    .partial_flag(True) \
-    .input(0, "x1", False, "required", "all") \
-    .input(1, "x2", False, "required", "all") \
-    .output(0, "y", True, "required", "all") \
-    .op_pattern("broadcast") \
-    .dtype_format(DataType.BOOL_None, DataType.BOOL_None, DataType.BOOL_None) \
-    .get_op_info()
-
-
-@op_info_register(logical_or_op_info)
-def _logical_or_ds_tbe():
-    """LogicalOr TBE register"""
-    return

mindspore/ops/_op_impl/tbe/logsoftmax.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.
-# ============================================================================
-
-"""LogSoftmax op"""
-from mindspore.ops.op_info_register import op_info_register, TBERegOp, DataType
-
-log_softmax_op_info = TBERegOp("LogSoftmax") \
-    .fusion_type("OPAQUE") \
-    .async_flag(False) \
-    .binfile_name("log_softmax.so") \
-    .compute_cost(10) \
-    .kernel_name("log_softmax_v2") \
-    .partial_flag(True) \
-    .attr("axis", "optional", "listInt", "all") \
-    .input(0, "logits", False, "required", "all") \
-    .output(0, "logsoftmax", 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(log_softmax_op_info)
-def _logsoftmax_tbe():
-    """LogSoftMaxGrad TBE register"""
-    return

mindspore/ops/_op_impl/tbe/logsoftmax_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.
-# ============================================================================
-
-"""LogSoftmax op"""
-from mindspore.ops.op_info_register import op_info_register, TBERegOp, DataType
-
-log_softmax_ds_op_info = TBERegOp("LogSoftmax") \
-    .fusion_type("OPAQUE") \
-    .async_flag(False) \
-    .binfile_name("log_softmax.so") \
-    .compute_cost(10) \
-    .kernel_name("log_softmax_v2") \
-    .partial_flag(True) \
-    .dynamic_shape(True) \
-    .attr("axis", "optional", "listInt", "all") \
-    .input(0, "logits", False, "required", "all") \
-    .output(0, "logsoftmax", 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(log_softmax_ds_op_info)
-def _logsoftmax_ds_tbe():
-    """LogSoftMaxGrad TBE register"""
-    return

mindspore/ops/_op_impl/tbe/logsoftmax_grad.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.
-# ============================================================================
-
-"""LogSoftmaxGrad op"""
-from mindspore.ops.op_info_register import op_info_register, TBERegOp, DataType
-
-log_softmax_grad_op_info = TBERegOp("LogSoftmaxGrad") \
-    .fusion_type("OPAQUE") \
-    .async_flag(False) \
-    .binfile_name("log_softmax_grad.so") \
-    .compute_cost(10) \
-    .kernel_name("log_softmax_grad") \
-    .partial_flag(True) \
-    .attr("axis", "optional", "listInt", "all", "-1") \
-    .input(0, "x", False, "required", "all") \
-    .input(1, "grad", 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) \
-    .get_op_info()
-
-
-@op_info_register(log_softmax_grad_op_info)
-def _logsoftmax_grad_tbe():
-    """LogSoftMaxGrad TBE register"""
-    return

mindspore/ops/_op_impl/tbe/logsoftmax_grad_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.
-# ============================================================================
-
-"""LogSoftmaxGrad op"""
-from mindspore.ops.op_info_register import op_info_register, TBERegOp, DataType
-
-log_softmax_grad_ds_op_info = TBERegOp("LogSoftmaxGrad") \
-    .fusion_type("OPAQUE") \
-    .async_flag(False) \
-    .binfile_name("log_softmax_grad.so") \
-    .compute_cost(10) \
-    .kernel_name("log_softmax_grad") \
-    .partial_flag(True) \
-    .dynamic_shape(True) \
-    .attr("axis", "optional", "listInt", "all", "-1") \
-    .input(0, "x", False, "required", "all") \
-    .input(1, "grad", 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) \
-    .get_op_info()
-
-
-@op_info_register(log_softmax_grad_ds_op_info)
-def _logsoftmax_grad_ds_tbe():
-    """LogSoftMaxGrad TBE register"""
-    return

mindspore/ops/_op_impl/tbe/lp_norm.py

@@ -1,40 +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.
-# ============================================================================
-
-"""LpNorm op"""
-from mindspore.ops.op_info_register import op_info_register, TBERegOp, DataType
-
-lp_norm_op_info = TBERegOp("LpNorm") \
-    .fusion_type("OPAQUE") \
-    .async_flag(False) \
-    .binfile_name("lp_norm.so") \
-    .compute_cost(10) \
-    .kernel_name("lp_norm") \
-    .partial_flag(True) \
-    .attr("p", "optional", "int", "all", "2") \
-    .attr("axis", "optional", "listInt", "all") \
-    .attr("keep_dims", "optional", "bool", "all", "false") \
-    .attr("epsilon", "optional", "float", "all", "1e-12") \
-    .input(0, "input", 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) \
-    .get_op_info()
-
-
-@op_info_register(lp_norm_op_info)
-def _lp_norm_tbe():
-    """LpNorm TBE register"""
-    return

mindspore/ops/_op_impl/tbe/lp_norm_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.
-# ============================================================================
-
-"""LpNorm op"""
-from mindspore.ops.op_info_register import op_info_register, TBERegOp, DataType
-
-lp_norm_dy_op_info = TBERegOp("LpNorm") \
-    .fusion_type("OPAQUE") \
-    .async_flag(False) \
-    .binfile_name("lp_norm.so") \
-    .compute_cost(10) \
-    .kernel_name("lp_norm") \
-    .partial_flag(True) \
-    .dynamic_shape(True) \
-    .attr("p", "optional", "int", "all", "2") \
-    .attr("axis", "optional", "listInt", "all") \
-    .attr("keep_dims", "optional", "bool", "all", "false") \
-    .attr("epsilon", "optional", "float", "all", "1e-12") \
-    .input(0, "input", 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) \
-    .get_op_info()
-
-
-@op_info_register(lp_norm_dy_op_info)
-def _lp_norm_ds_tbe():
-    """LpNorm Dynamic Shape TBE register"""
-    return

mindspore/ops/_op_impl/tbe/lrn.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.
-# ============================================================================
-
-"""LRN op"""
-from mindspore.ops.op_info_register import op_info_register, TBERegOp, DataType
-
-lrn_op_info = TBERegOp("LRN") \
-    .fusion_type("OPAQUE") \
-    .async_flag(False) \
-    .binfile_name("lrn.so") \
-    .compute_cost(10) \
-    .kernel_name("lrn") \
-    .partial_flag(True) \
-    .attr("depth_radius", "optional", "int", "all", "5") \
-    .attr("bias", "optional", "float", "all", "1.0") \
-    .attr("alpha", "optional", "float", "all", "1.0") \
-    .attr("beta", "optional", "float", "all", "0.5") \
-    .attr("norm_region", "optional", "str", "all", "ACROSS_CHANNELS") \
-    .input(0, "x", False, "required", "all") \
-    .output(0, "y", False, "required", "all") \
-    .dtype_format(DataType.F16_Default, DataType.F16_Default) \
-    .dtype_format(DataType.F32_Default, DataType.F32_Default) \
-    .get_op_info()
-
-
-@op_info_register(lrn_op_info)
-def _lrn_tbe():
-    """LRN TBE register"""
-    return

mindspore/ops/_op_impl/tbe/lrn_grad.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.
-# ============================================================================
-
-"""LRNGrad op"""
-from mindspore.ops.op_info_register import op_info_register, TBERegOp, DataType
-
-lrn_grad_op_info = TBERegOp("LRNGrad") \
-    .fusion_type("OPAQUE") \
-    .async_flag(False) \
-    .binfile_name("lrn_grad.so") \
-    .compute_cost(10) \
-    .kernel_name("lrn_grad") \
-    .partial_flag(True) \
-    .attr("depth_radius", "optional", "int", "all") \
-    .attr("bias", "optional", "float", "all") \
-    .attr("alpha", "optional", "float", "all") \
-    .attr("beta", "optional", "float", "all") \
-    .input(0, "grads", False, "required", "all") \
-    .input(1, "x", False, "required", "all") \
-    .input(2, "y", False, "required", "all") \
-    .output(0, "z", False, "required", "all") \
-    .dtype_format(DataType.F16_Default, DataType.F16_Default, DataType.F16_Default, DataType.F16_Default) \
-    .dtype_format(DataType.F32_Default, DataType.F32_Default, DataType.F32_Default, DataType.F32_Default) \
-    .get_op_info()
-
-
-@op_info_register(lrn_grad_op_info)
-def _lrn_grad_tbe():
-    """LRNGrad TBE register"""
-    return

mindspore/ops/_op_impl/tbe/lstm_input_grad.py

@@ -1,51 +0,0 @@
-# Copyright 2020 Huawei Technologies Co., Ltd
-#
-# Licensed under the Apache License, Version 2.0 (the "License");
-# you may not use this file except in compliance with the License.
-# You may obtain a copy of the License at
-#
-# http://www.apache.org/licenses/LICENSE-2.0
-#
-# Unless required by applicable law or agreed to in writing, software
-# distributed under the License is distributed on an "AS IS" BASIS,
-# WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
-# See the License for the specific language governing permissions and
-# limitations under the License.
-# ============================================================================
-
-"""LSTMInputGrad op"""
-from mindspore.ops.op_info_register import op_info_register, TBERegOp, DataType
-
-lstm_input_grad_op_info = TBERegOp("LSTMInputGrad") \
-    .fusion_type("OPAQUE") \
-    .async_flag(False) \
-    .binfile_name("lstm_input_grad.so") \
-    .compute_cost(10) \
-    .kernel_name("lstm_input_grad") \
-    .partial_flag(True) \
-    .input(0, "w", False, "required", "all") \
-    .input(1, "init_c", False, "required", "all") \
-    .input(2, "c", False, "required", "all") \
-    .input(3, "dy", False, "required", "all") \
-    .input(4, "dh", False, "required", "all") \
-    .input(5, "dc", False, "required", "all") \
-    .input(6, "i", False, "required", "all") \
-    .input(7, "j", False, "required", "all") \
-    .input(8, "f", False, "required", "all") \
-    .input(9, "o", False, "required", "all") \
-    .input(10, "tanhct", False, "optional", "all") \
-    .output(0, "dx", False, "required", "all") \
-    .output(1, "dh_prev", False, "required", "all") \
-    .output(2, "dc_prev", False, "required", "all") \
-    .output(3, "dgate", False, "required", "all") \
-    .dtype_format(DataType.F16_FracNZ, DataType.F16_FracNZ, DataType.F16_FracNZ, DataType.F16_FracNZ,
-                  DataType.F16_FracNZ, DataType.F16_FracNZ, DataType.F16_FracNZ, DataType.F16_FracNZ,
-                  DataType.F16_FracNZ, DataType.F16_FracNZ, DataType.F16_FracNZ, DataType.F16_FracNZ,
-                  DataType.F16_FracNZ, DataType.F16_FracNZ, DataType.F16_FracNZ) \
-    .get_op_info()
-
-
-@op_info_register(lstm_input_grad_op_info)
-def _lstm_input_grad_tbe():
-    """LSTMInputGrad TBE register"""
-    return

mindspore/ops/_op_impl/tbe/masked_fill.py

@@ -1,40 +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.
-# ============================================================================
-
-"""MaskedFill op"""
-from mindspore.ops.op_info_register import op_info_register, TBERegOp, DataType
-
-masked_fill_op_info = TBERegOp("MaskedFill") \
-    .fusion_type("OPAQUE") \
-    .async_flag(False) \
-    .binfile_name("masked_fill.so") \
-    .compute_cost(10) \
-    .kernel_name("masked_fill") \
-    .partial_flag(True) \
-    .input(0, "input", False, "required", "all") \
-    .input(1, "mask", False, "required", "all") \
-    .input(2, "value", False, "required", "all") \
-    .output(0, "output", False, "required", "all") \
-    .dtype_format(DataType.F16_Default, DataType.BOOL_Default, DataType.F16_Default, DataType.F16_Default,) \
-    .dtype_format(DataType.F32_Default, DataType.BOOL_Default, DataType.F32_Default, DataType.F32_Default,) \
-    .dtype_format(DataType.I8_Default, DataType.BOOL_Default, DataType.I8_Default, DataType.I8_Default,) \
-    .dtype_format(DataType.I32_Default, DataType.BOOL_Default, DataType.I32_Default, DataType.I32_Default,) \
-    .get_op_info()
-
-
-@op_info_register(masked_fill_op_info)
-def _masked_fill_tbe():
-    """MaskedFill TBE register"""
-    return

mindspore/ops/_op_impl/tbe/masked_fill_ds.py

@@ -1,41 +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.
-# ============================================================================
-
-"""MaskedFill op"""
-from mindspore.ops.op_info_register import op_info_register, TBERegOp, DataType
-
-masked_fill_op_info = TBERegOp("MaskedFill") \
-    .fusion_type("OPAQUE") \
-    .async_flag(False) \
-    .binfile_name("masked_fill.so") \
-    .compute_cost(10) \
-    .kernel_name("masked_fill") \
-    .partial_flag(True) \
-    .dynamic_shape(True) \
-    .input(0, "input", False, "required", "all") \
-    .input(1, "mask", False, "required", "all") \
-    .input(2, "value", False, "required", "all") \
-    .output(0, "output", False, "required", "all") \
-    .dtype_format(DataType.F16_Default, DataType.BOOL_Default, DataType.F16_Default, DataType.F16_Default,) \
-    .dtype_format(DataType.F32_Default, DataType.BOOL_Default, DataType.F32_Default, DataType.F32_Default,) \
-    .dtype_format(DataType.I8_Default, DataType.BOOL_Default, DataType.I8_Default, DataType.I8_Default,) \
-    .dtype_format(DataType.I32_Default, DataType.BOOL_Default, DataType.I32_Default, DataType.I32_Default,) \
-    .get_op_info()
-
-
-@op_info_register(masked_fill_op_info)
-def _masked_fill_ds_tbe():
-    """MaskedFill TBE register"""
-    return