mindspore 2.4.0__cp311-cp311-macosx_10_15_x86_64.whl

mindspore/ops/_op_impl/akg/gpu/__init__.py

@@ -0,0 +1,24 @@
+# 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.
+
+"""__init__"""
+from .coo2csr import _coo2csr_akg
+from .csr2coo import _csr2coo_akg
+from .csr_gather import _csr_gather_akg
+from .csr_div import _csr_div_akg
+from .csr_mul import _csr_mul_akg
+from .csr_mv import _csr_mv_akg
+from .csr_mm import _csr_mm_akg
+from .csr_reduce_sum import _csr_reduce_sum_akg
+# Please insert op register in lexicographical order of the filename.

mindspore/ops/_op_impl/akg/gpu/coo2csr.py

@@ -0,0 +1,29 @@
+# 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.
+
+"""COO2CSR op"""
+from mindspore.ops.op_info_register import op_info_register, AkgGpuRegOp, DataType
+
+coo2csr_op_info = AkgGpuRegOp("COO2CSR") \
+    .fusion_type("OPAQUE") \
+    .input(0, "row_indices") \
+    .output(0, "output") \
+    .dtype_format(DataType.I64_Default, DataType.I64_Default) \
+    .dtype_format(DataType.I32_Default, DataType.I32_Default) \
+    .get_op_info()
+
+@op_info_register(coo2csr_op_info)
+def _coo2csr_akg():
+    """COO2CSR AutoDiff register"""
+    return

mindspore/ops/_op_impl/akg/gpu/csr2coo.py

@@ -0,0 +1,29 @@
+# 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.
+
+"""CSR2COO op"""
+from mindspore.ops.op_info_register import op_info_register, AkgGpuRegOp, DataType
+
+csr2coo_op_info = AkgGpuRegOp("CSR2COO") \
+    .fusion_type("OPAQUE") \
+    .input(0, "indptr") \
+    .output(0, "output") \
+    .dtype_format(DataType.I64_Default, DataType.I64_Default) \
+    .dtype_format(DataType.I32_Default, DataType.I32_Default) \
+    .get_op_info()
+
+@op_info_register(csr2coo_op_info)
+def _csr2coo_akg():
+    """CSR2COO AutoDiff register"""
+    return

mindspore/ops/_op_impl/akg/gpu/csr_div.py

@@ -0,0 +1,36 @@
+# 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.
+
+"""CSRDiv op"""
+from mindspore.ops.op_info_register import op_info_register, AkgGpuRegOp, DataType
+
+csr_div_op_info = AkgGpuRegOp("CSRDiv") \
+    .fusion_type("OPAQUE") \
+    .input(0, "indptr") \
+    .input(1, "indices") \
+    .input(2, "values") \
+    .input(4, "dense_tensor") \
+    .output(0, "output0") \
+    .dtype_format(DataType.I64_Default, DataType.I64_Default, DataType.F32_Default, \
+                  DataType.F32_Default, \
+                  DataType.F32_Default) \
+    .dtype_format(DataType.I32_Default, DataType.I32_Default, DataType.F32_Default, \
+                  DataType.F32_Default, \
+                  DataType.F32_Default) \
+    .get_op_info()
+
+@op_info_register(csr_div_op_info)
+def _csr_div_akg():
+    """CSRDiv AutoDiff register"""
+    return

mindspore/ops/_op_impl/akg/gpu/csr_gather.py

@@ -0,0 +1,33 @@
+# 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.
+
+"""CSRGatherop"""
+from mindspore.ops.op_info_register import op_info_register, AkgGpuRegOp, DataType
+
+csr_gather_op_info = AkgGpuRegOp("CSRGather") \
+    .fusion_type("OPAQUE") \
+    .input(0, "indptr") \
+    .input(1, "indices") \
+    .input(2, "dense") \
+    .output(0, "output") \
+    .dtype_format(DataType.I64_Default, DataType.I64_Default, DataType.F32_Default, \
+                  DataType.F32_Default) \
+    .dtype_format(DataType.I32_Default, DataType.I32_Default, DataType.F32_Default, \
+                  DataType.F32_Default) \
+    .get_op_info()
+
+@op_info_register(csr_gather_op_info)
+def _csr_gather_akg():
+    """CSRGather AutoDiff register"""
+    return

mindspore/ops/_op_impl/akg/gpu/csr_mm.py

@@ -0,0 +1,37 @@
+# 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.
+
+"""CSRMM op"""
+from mindspore.ops.op_info_register import op_info_register, AkgGpuRegOp, DataType
+
+csr_mm_op_info = AkgGpuRegOp("CSRMM") \
+    .fusion_type("OPAQUE") \
+    .input(0, "indptr") \
+    .input(1, "indices") \
+    .input(2, "values") \
+    .input(4, "dense_tensor") \
+    .output(0, "output") \
+    .dtype_format(DataType.I32_Default, DataType.I32_Default, DataType.F32_Default, \
+                  DataType.F32_Default, \
+                  DataType.F32_Default) \
+    .dtype_format(DataType.I64_Default, DataType.I64_Default, DataType.F32_Default, \
+                  DataType.F32_Default, \
+                  DataType.F32_Default) \
+    .get_op_info()
+
+
+@op_info_register(csr_mm_op_info)
+def _csr_mm_akg():
+    """CSRMM AutoDiff register"""
+    return

mindspore/ops/_op_impl/akg/gpu/csr_mul.py

@@ -0,0 +1,36 @@
+# 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.
+
+"""CSRMul op"""
+from mindspore.ops.op_info_register import op_info_register, AkgGpuRegOp, DataType
+
+csr_mul_op_info = AkgGpuRegOp("CSRMul") \
+    .fusion_type("OPAQUE") \
+    .input(0, "indptr") \
+    .input(1, "indices") \
+    .input(2, "values") \
+    .input(4, "dense_tensor") \
+    .output(0, "output0") \
+    .dtype_format(DataType.I64_Default, DataType.I64_Default, DataType.F32_Default, \
+                  DataType.F32_Default, \
+                  DataType.F32_Default) \
+    .dtype_format(DataType.I32_Default, DataType.I32_Default, DataType.F32_Default, \
+                  DataType.F32_Default, \
+                  DataType.F32_Default) \
+    .get_op_info()
+
+@op_info_register(csr_mul_op_info)
+def _csr_mul_akg():
+    """CSRMul AutoDiff register"""
+    return

mindspore/ops/_op_impl/akg/gpu/csr_mv.py

@@ -0,0 +1,36 @@
+# 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.
+
+"""CSRMV op"""
+from mindspore.ops.op_info_register import op_info_register, AkgGpuRegOp, DataType
+
+csr_mv_op_info = AkgGpuRegOp("CSRMV") \
+    .fusion_type("OPAQUE") \
+    .input(0, "indptr") \
+    .input(1, "indices") \
+    .input(2, "values") \
+    .input(4, "dense_tensor") \
+    .output(0, "output") \
+    .dtype_format(DataType.I32_Default, DataType.I32_Default, DataType.F32_Default, \
+                  DataType.F32_Default, \
+                  DataType.F32_Default) \
+    .dtype_format(DataType.I64_Default, DataType.I64_Default, DataType.F32_Default, \
+                  DataType.F32_Default, \
+                  DataType.F32_Default) \
+    .get_op_info()
+
+@op_info_register(csr_mv_op_info)
+def _csr_mv_akg():
+    """CSRMV AutoDiff register"""
+    return

mindspore/ops/_op_impl/akg/gpu/csr_reduce_sum.py

@@ -0,0 +1,33 @@
+# 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.
+
+"""CSRReduceSum op"""
+from mindspore.ops.op_info_register import op_info_register, AkgGpuRegOp, DataType
+
+csr_reduce_sum_op_info = AkgGpuRegOp("CSRReduceSum") \
+    .fusion_type("OPAQUE") \
+    .input(0, "indptr") \
+    .input(1, "indices") \
+    .input(2, "values") \
+    .output(0, "output") \
+    .dtype_format(DataType.I64_Default, DataType.I64_Default, DataType.F32_Default, \
+                  DataType.F32_Default) \
+    .dtype_format(DataType.I32_Default, DataType.I32_Default, DataType.F32_Default, \
+                  DataType.F32_Default) \
+    .get_op_info()
+
+@op_info_register(csr_reduce_sum_op_info)
+def _csr_reduce_sum_akg():
+    """CSRReduceSum AutoDiff register"""
+    return

mindspore/ops/_op_impl/cpu/__init__.py

@@ -0,0 +1,78 @@
+# Copyright 2021-2023 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.
+
+"""cpu ops"""
+from .cast import _cast_cpu
+from .mul import _mul_cpu
+from .sub import _sub_cpu
+from .pow import _pow_cpu
+from .real_div import _real_div_cpu
+from .div import _div_cpu
+from .concat_offset import _concat_offset_cpu
+from .dynamic_shape import _dynamic_shape_cpu
+from .dynamic_stitch import _dynamic_stitch_cpu
+from .split import _split_cpu
+from .adam import _adam_cpu
+from .adam_weight_decay import _adam_weight_decay_cpu
+from .arg_max import _arg_max_cpu
+from .arg_min_with_value import _arg_min_with_value_cpu
+from .arg_max_with_value import _arg_max_with_value_cpu
+from .dropout import _dropout_cpu
+from .dropout_grad import _dropout_grad_cpu
+from .gather_d import _gather_cpu
+from .gather_d_grad import _gather_d_grad_cpu
+from .gather_d_grad_v2 import _gather_d_grad_v2_cpu
+from .gather_v2 import _gather_v2_cpu
+from .maximum import _maximum_cpu
+from .maximum_grad import _maximum_grad_cpu
+from .conv2d import _conv2d_cpu
+from .conv3d import _conv3d_cpu
+from .hsigmoid import _hsigmoid_cpu
+from .hsigmoid_grad import _hsigmoid_grad_cpu
+from .hswish import _hswish_cpu
+from .hswish_grad import _hswish_grad_cpu
+from .identity_n import _identity_n_cpu
+from .is_finite import _is_finite_cpu
+from .layer_norm import _layer_norm_cpu
+from .layer_norm_grad import _layer_norm_grad_cpu
+from .minimum import _minimum_cpu
+from .minimum_grad import _minimum_grad_cpu
+from .equal_count import _equal_count_cpu
+from .mirror_pad import _mirror_pad_cpu
+from .mirror_pad_grad import _mirror_pad_grad_cpu
+from .stack import _stack_cpu
+from .reduce_mean import _reduce_mean_cpu
+from .reduce_max import _reduce_max_cpu
+from .reduce_sum import _reduce_sum_cpu
+from .reduce_min import _reduce_min_cpu
+from .reduce_prod import _reduce_prod_cpu
+from .reduce_all import _reduce_all_cpu
+from .reduce_any import _reduce_any_cpu
+from .transpose import _transpose_cpu
+from .tile import _tile_cpu
+from .top_k import _top_k_cpu
+from .one_hot import _one_hot_cpu
+from .pad import _pad_cpu
+from .range import _range_cpu
+from .tensor_copy_slices import _tensor_copy_slices_cpu
+from .l2loss import _l2loss_cpu
+from .pyexecute import _pyexecute_cpu
+from .pyfunc import _pyfunc_cpu
+from .buffer_append import _buffer_append_cpu
+from .buffer_get import _buffer_get_cpu
+from .buffer_sample import _buffer_sample_cpu
+from .priority_replay_buffer import _prb_push_op_cpu
+from .priority_replay_buffer import _prb_sample_op_cpu
+from .space_to_batch_nd import _space_to_batch_nd_cpu
+from .sspaddmm import _sspaddmm_cpu

mindspore/ops/_op_impl/cpu/adam.py

@@ -0,0 +1,49 @@
+# Copyright 2020-2023 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.
+# ============================================================================
+
+"""Adam op"""
+from mindspore.ops.op_info_register import op_info_register, CpuRegOp, DataType
+
+adam_op_info = CpuRegOp("Adam") \
+    .input(0, "var", "required") \
+    .input(1, "m", "required") \
+    .input(2, "v", "required") \
+    .input(3, "beta1_power", "required") \
+    .input(4, "beta2_power", "required") \
+    .input(5, "lr", "required") \
+    .input(6, "beta1", "required") \
+    .input(7, "beta2", "required") \
+    .input(8, "epsilon", "required") \
+    .input(9, "gradient", "required") \
+    .output(0, "output0", "required") \
+    .output(1, "output0", "required") \
+    .output(2, "output0", "required") \
+    .dtype_format(DataType.F32_Default, DataType.F32_Default, DataType.F32_Default,
+                  DataType.F32_Default, DataType.F32_Default, DataType.F32_Default,
+                  DataType.F32_Default, DataType.F32_Default, DataType.F32_Default,
+                  DataType.F32_Default, DataType.F32_Default, DataType.F32_Default,
+                  DataType.F32_Default) \
+    .dtype_format(DataType.F16_Default, DataType.F16_Default, DataType.F16_Default,
+                  DataType.F32_Default, DataType.F32_Default, DataType.F32_Default,
+                  DataType.F32_Default, DataType.F32_Default, DataType.F32_Default,
+                  DataType.F16_Default, DataType.F16_Default, DataType.F16_Default,
+                  DataType.F16_Default) \
+    .get_op_info()
+
+
+@op_info_register(adam_op_info)
+def _adam_cpu():
+    """Adam cpu register"""
+    return

mindspore/ops/_op_impl/cpu/adam_weight_decay.py

@@ -0,0 +1,47 @@
+# 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.
+# ============================================================================
+
+"""AdamWeightDecay op"""
+from mindspore.ops.op_info_register import op_info_register, CpuRegOp, DataType
+
+adam_weight_decay_op_info = CpuRegOp("AdamWeightDecay") \
+    .input(0, "var", "required") \
+    .input(1, "m", "required") \
+    .input(2, "v", "required") \
+    .input(3, "lr", "required") \
+    .input(4, "beta1", "required") \
+    .input(5, "beta2", "required") \
+    .input(6, "epsilon", "required") \
+    .input(7, "decay", "required") \
+    .input(8, "gradient", "required") \
+    .input(9, "use_locking", "required") \
+    .output(0, "output0", "required") \
+    .output(1, "output1", "required") \
+    .output(2, "output2", "required") \
+    .dtype_format(DataType.F32_Default, DataType.F32_Default, DataType.F32_Default,
+                  DataType.F32_Default, DataType.F32_Default, DataType.F32_Default,
+                  DataType.F32_Default, DataType.F32_Default, DataType.F32_Default, DataType.BOOL_Default,
+                  DataType.F32_Default, DataType.F32_Default, DataType.F32_Default) \
+    .dtype_format(DataType.F16_Default, DataType.F16_Default, DataType.F16_Default,
+                  DataType.F32_Default, DataType.F32_Default, DataType.F32_Default,
+                  DataType.F32_Default, DataType.F32_Default, DataType.F16_Default, DataType.BOOL_Default,
+                  DataType.F16_Default, DataType.F16_Default, DataType.F16_Default) \
+    .get_op_info()
+
+
+@op_info_register(adam_weight_decay_op_info)
+def _adam_weight_decay_cpu():
+    """AdamWeightDecay cpu register"""
+    return

mindspore/ops/_op_impl/cpu/arg_max.py

@@ -0,0 +1,30 @@
+# 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.
+# ============================================================================
+
+"""Argmax op"""
+from mindspore.ops.op_info_register import op_info_register, CpuRegOp, DataType
+
+arg_max_op_info = CpuRegOp("Argmax") \
+    .input(0, "x", "required") \
+    .output(0, "y", "required") \
+    .dtype_format(DataType.F32_Default, DataType.I32_Default) \
+    .dtype_format(DataType.F16_Default, DataType.I32_Default) \
+    .get_op_info()
+
+
+@op_info_register(arg_max_op_info)
+def _arg_max_cpu():
+    """Argmax cpu register"""
+    return

mindspore/ops/_op_impl/cpu/arg_max_with_value.py

@@ -0,0 +1,31 @@
+# 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.
+# ============================================================================
+
+"""ArgMaxWithValue op"""
+from mindspore.ops.op_info_register import op_info_register, CpuRegOp, DataType
+
+arg_max_with_value_op_info = CpuRegOp("ArgMaxWithValue") \
+    .input(0, "x", "required") \
+    .output(0, "indice", "required") \
+    .output(1, "values", "required") \
+    .dtype_format(DataType.F32_Default, DataType.I32_Default, DataType.F32_Default) \
+    .dtype_format(DataType.F16_Default, DataType.I32_Default, DataType.F16_Default) \
+    .get_op_info()
+
+
+@op_info_register(arg_max_with_value_op_info)
+def _arg_max_with_value_cpu():
+    """ArgMaxWithValue cpu register"""
+    return

mindspore/ops/_op_impl/cpu/arg_min_with_value.py

@@ -0,0 +1,31 @@
+# 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.
+# ============================================================================
+
+"""ArgMinWithValue op"""
+from mindspore.ops.op_info_register import op_info_register, CpuRegOp, DataType
+
+arg_min_with_value_op_info = CpuRegOp("ArgMinWithValue") \
+    .input(0, "x", "required") \
+    .output(0, "indice", "required") \
+    .output(1, "values", "required") \
+    .dtype_format(DataType.F32_Default, DataType.I32_Default, DataType.F32_Default) \
+    .dtype_format(DataType.F16_Default, DataType.I32_Default, DataType.F16_Default) \
+    .get_op_info()
+
+
+@op_info_register(arg_min_with_value_op_info)
+def _arg_min_with_value_cpu():
+    """ArgMinWithValue cpu register"""
+    return

mindspore/ops/_op_impl/cpu/buffer_append.py

@@ -0,0 +1,28 @@
+# 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.
+
+"""BufferAppend op"""
+from mindspore.ops.op_info_register import op_info_register, CpuRegOp, DataType
+
+buffer_append_op_info = CpuRegOp("BufferAppend") \
+    .input(0, "x", "dynamic") \
+    .output(0, "output", "dynamic") \
+    .dtype_format(DataType.I32_Default, DataType.I32_Default) \
+    .get_op_info()
+
+
+@op_info_register(buffer_append_op_info)
+def _buffer_append_cpu():
+    """BufferAppend cpu register"""
+    return

mindspore/ops/_op_impl/cpu/buffer_get.py

@@ -0,0 +1,28 @@
+# 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.
+
+"""BufferGetItem op"""
+from mindspore.ops.op_info_register import op_info_register, CpuRegOp, DataType
+
+buffer_get_op_info = CpuRegOp("BufferGetItem") \
+    .input(0, "x", "dynamic") \
+    .output(0, "output", "dynamic") \
+    .dtype_format(DataType.I32_Default, DataType.I32_Default) \
+    .get_op_info()
+
+
+@op_info_register(buffer_get_op_info)
+def _buffer_get_cpu():
+    """BufferGetItem cpu register"""
+    return

mindspore/ops/_op_impl/cpu/buffer_sample.py

@@ -0,0 +1,28 @@
+# 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.
+
+"""BufferSample op"""
+from mindspore.ops.op_info_register import op_info_register, CpuRegOp, DataType
+
+buffer_sample_op_info = CpuRegOp("BufferSample") \
+    .input(0, "x", "dynamic") \
+    .output(0, "output", "dynamic") \
+    .dtype_format(DataType.I32_Default, DataType.I32_Default) \
+    .get_op_info()
+
+
+@op_info_register(buffer_sample_op_info)
+def _buffer_sample_cpu():
+    """BufferSample cpu register"""
+    return