mindspore 2.7.0rc1__cp310-cp310-win_amd64.whl → 2.7.1__cp310-cp310-win_amd64.whl

mindspore/nn/reinforcement/tensor_array.py

@@ -1,145 +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.
-# ============================================================================
-"""
-TensorArray
-"""
-from __future__ import absolute_import
-
-from mindspore.nn.cell import Cell
-from mindspore.ops.operations import _tensor_array as ta
-from mindspore import _checkparam as Validator
-from mindspore.common import dtype as mstype
-
-
-class TensorArray(Cell):
-    r"""TensorArray: a dynamic array to store tensors.
-
-    .. warning::
-        This is an experiential prototype that is subject to change and/or deletion.
-
-    Args:
-        dtype (mindspore.dtype): the data type in the TensorArray.
-        element_shape (tuple[int]): the shape of each tensor in a TensorArray.
-        dynamic_size (bool): if ``true`` , the size of TensorArray can be increased. Default: ``True`` .
-        size (int): if dynamic_size=False, `size` means the max_size of the TensorArray.
-        name (str): the name of this TensorArray. Default: ``"TA"`` .
-
-    Supported Platforms:
-        ``GPU`` ``CPU``
-
-    Examples:
-        >>> import mindspore
-        >>> import mindspore.nn as nn
-        >>> ta = nn.TensorArray(mindspore.int64, ())
-        >>> ta.write(0, 1)
-        >>> ta.write(1, 2)
-        >>> ans = ta.read(1)
-        >>> print(ans)
-        2
-        >>> s = ta.stack()
-        >>> print(s)
-        [1 2]
-        >>> ta.clear()
-        >>> ta.write(0, 3)
-        >>> ans = ta.read(0)
-        >>> print(ans)
-        3
-        >>> ta.close()
-    """
-    def __init__(self, dtype, element_shape, dynamic_size=True, size=0, name="TA"):
-        """Initialize TensorArray"""
-        super(TensorArray, self).__init__()
-        Validator.check_subclass("dtype", dtype, mstype.number_type + (mstype.bool_,), self.cls_name)
-        Validator.check_int(size, 0, Validator.GE, "size", self.cls_name)
-        self.handle_ = ta.TensorArray(dtype, element_shape, dynamic_size, size, name)()
-        self.tensor_array_write = ta.TensorArrayWrite()
-        self.tensor_array_read = ta.TensorArrayRead(dtype, element_shape)
-        self.tensor_array_close = ta.TensorArrayClose()
-        self.tensor_array_clear = ta.TensorArrayClear()
-        self.tensor_array_stack = ta.TensorArrayStack(dtype, element_shape, dynamic_size, size)
-        self.tensor_array_size = ta.TensorArraySize()
-
-    def write(self, index, value):
-        """
-        Write value(Tensor) to TensorArray in position index.
-
-        Args:
-            index ([int, mindspore.int64]): The position to write.
-            value (Tensor): The value to add into the TensorArray.
-
-        Returns:
-            Bool, true.
-        """
-        self.tensor_array_write(self.handle_, index, value)
-        return True
-
-    def read(self, index):
-        """
-        Read tensor form the TensorArray by the given position index.
-
-        Args:
-            index ([int, mindspore.int64]): The given index to get the tensor.
-
-        Returns:
-            Tensor, the value in position index.
-        """
-        value = self.tensor_array_read(self.handle_, index)
-        return value
-
-    def close(self):
-        """
-        Close the created TensorArray.
-
-        .. warning::
-            Once close the TensorArray, every functions belong to this TensorArray will be disaviliable.
-            Every resources created in TensorArray will be removed. If this TensorArray will be used in next step
-            or somewhere, eg: next loop, please use `clear` instead.
-
-        Returns:
-            Bool, true.
-        """
-        self.tensor_array_close(self.handle_)
-        return True
-
-    def clear(self):
-        """
-        Clear the created TensorArray. Only reset the TensorArray, clear the data and reset the size
-        in TensorArray and keep the instance of this TensorArray.
-
-        Returns:
-            Bool, true.
-        """
-        self.tensor_array_clear(self.handle_)
-        return True
-
-    def stack(self):
-        """
-        Stack the values in TensorArray into a stacked Tensor.
-
-        Returns:
-            Tensor, all the values will be stacked into one tensor.
-        """
-        ans = self.tensor_array_stack(self.handle_)
-        return ans
-
-    def size(self):
-        """
-        The logical size of TensorArray.
-
-        Returns:
-            Tensor, the size of TensorArray.
-        """
-        size = self.tensor_array_size(self.handle_)
-        return size

mindspore/ops/_op_impl/aicpu/priority_replay_buffer.py

@@ -1,113 +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.
-# ============================================================================
-
-"""PriorityReplayBuffer op"""
-from mindspore.ops.op_info_register import op_info_register, AiCPURegOp, DataType
-
-
-prb_create_op_info = AiCPURegOp("PriorityReplayBufferCreate") \
-    .fusion_type("OPAQUE") \
-    .output(0, "handle", "required") \
-    .attr("capacity", "int") \
-    .attr("alpha", "float") \
-    .attr("beta", "float") \
-    .attr("schema", "listInt") \
-    .attr("seed0", "int") \
-    .attr("seed1", "int") \
-    .dtype_format(DataType.I64_Default) \
-    .get_op_info()
-
-
-prb_push_op_info = AiCPURegOp("PriorityReplayBufferPush") \
-    .input(0, "transition", "dynamic") \
-    .output(0, "handle", "required") \
-    .attr("handle", "int") \
-    .dtype_format(DataType.BOOL_Default, DataType.I64_Default) \
-    .dtype_format(DataType.I8_Default, DataType.I64_Default) \
-    .dtype_format(DataType.I16_Default, DataType.I64_Default) \
-    .dtype_format(DataType.I32_Default, DataType.I64_Default) \
-    .dtype_format(DataType.I64_Default, DataType.I64_Default) \
-    .dtype_format(DataType.F16_Default, DataType.I64_Default) \
-    .dtype_format(DataType.U8_Default, DataType.I64_Default) \
-    .dtype_format(DataType.U16_Default, DataType.I64_Default) \
-    .dtype_format(DataType.U32_Default, DataType.I64_Default) \
-    .dtype_format(DataType.U64_Default, DataType.I64_Default) \
-    .dtype_format(DataType.F32_Default, DataType.I64_Default) \
-    .get_op_info()
-
-
-prb_sample_op_info = AiCPURegOp("PriorityReplayBufferSample") \
-    .output(0, "transitions", "dynamic") \
-    .attr("handle", "int") \
-    .attr("batch_size", "int") \
-    .attr("schema", "listInt") \
-    .dtype_format(DataType.BOOL_Default) \
-    .dtype_format(DataType.I8_Default) \
-    .dtype_format(DataType.I16_Default) \
-    .dtype_format(DataType.I32_Default) \
-    .dtype_format(DataType.I64_Default) \
-    .dtype_format(DataType.F16_Default) \
-    .dtype_format(DataType.U8_Default) \
-    .dtype_format(DataType.U16_Default) \
-    .dtype_format(DataType.U32_Default) \
-    .dtype_format(DataType.U64_Default) \
-    .dtype_format(DataType.F32_Default) \
-    .get_op_info()
-
-
-prb_update_op_info = AiCPURegOp("PriorityReplayBufferUpdate") \
-    .input(0, "indices", "require") \
-    .input(1, "priorities", "require") \
-    .output(0, "handle", "require") \
-    .attr("handle", "int") \
-    .dtype_format(DataType.I64_Default, DataType.F32_Default, DataType.I64_Default) \
-    .get_op_info()
-
-
-prb_destroy_op_info = AiCPURegOp("PriorityReplayBufferDestroy") \
-    .output(0, "handle", "required") \
-    .attr("handle", "int") \
-    .dtype_format(DataType.I64_Default) \
-    .get_op_info()
-
-
-@op_info_register(prb_create_op_info)
-def _prb_create_op_cpu():
-    """PriorityReplayBufferSample AICPU register"""
-    return
-
-
-@op_info_register(prb_push_op_info)
-def _prb_push_op_cpu():
-    """PriorityReplayBufferPush AICPU register"""
-    return
-
-
-@op_info_register(prb_sample_op_info)
-def _prb_sample_op_cpu():
-    """PriorityReplayBufferSample AICPU register"""
-    return
-
-
-@op_info_register(prb_update_op_info)
-def _prb_update_op_cpu():
-    """PriorityReplayBufferUpdate AICPU register"""
-    return
-
-
-@op_info_register(prb_destroy_op_info)
-def _prb_destroy_op_cpu():
-    """PriorityReplayBufferDestroy AICPU register"""
-    return

mindspore/ops/_op_impl/aicpu/reservoir_replay_buffer.py

@@ -1,96 +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.
-# ============================================================================
-
-"""ReservoirReplayBuffer op"""
-from mindspore.ops.op_info_register import op_info_register, AiCPURegOp, DataType
-
-
-rrb_create_op_info = AiCPURegOp("ReservoirReplayBufferCreate") \
-    .fusion_type("OPAQUE") \
-    .output(0, "handle", "required") \
-    .attr("capacity", "int") \
-    .attr("schema", "listInt") \
-    .attr("seed0", "int") \
-    .attr("seed1", "int") \
-    .dtype_format(DataType.I64_Default) \
-    .get_op_info()
-
-
-rrb_push_op_info = AiCPURegOp("ReservoirReplayBufferPush") \
-    .input(0, "transition", "dynamic") \
-    .output(0, "handle", "required") \
-    .attr("handle", "int") \
-    .dtype_format(DataType.BOOL_Default, DataType.I64_Default) \
-    .dtype_format(DataType.I8_Default, DataType.I64_Default) \
-    .dtype_format(DataType.I16_Default, DataType.I64_Default) \
-    .dtype_format(DataType.I32_Default, DataType.I64_Default) \
-    .dtype_format(DataType.I64_Default, DataType.I64_Default) \
-    .dtype_format(DataType.F16_Default, DataType.I64_Default) \
-    .dtype_format(DataType.U8_Default, DataType.I64_Default) \
-    .dtype_format(DataType.U16_Default, DataType.I64_Default) \
-    .dtype_format(DataType.U32_Default, DataType.I64_Default) \
-    .dtype_format(DataType.U64_Default, DataType.I64_Default) \
-    .dtype_format(DataType.F32_Default, DataType.I64_Default) \
-    .get_op_info()
-
-
-rrb_sample_op_info = AiCPURegOp("ReservoirReplayBufferSample") \
-    .output(0, "transitions", "dynamic") \
-    .attr("handle", "int") \
-    .attr("batch_size", "int") \
-    .attr("schema", "listInt") \
-    .dtype_format(DataType.BOOL_Default) \
-    .dtype_format(DataType.I8_Default) \
-    .dtype_format(DataType.I16_Default) \
-    .dtype_format(DataType.I32_Default) \
-    .dtype_format(DataType.I64_Default) \
-    .dtype_format(DataType.F16_Default) \
-    .dtype_format(DataType.U8_Default) \
-    .dtype_format(DataType.U16_Default) \
-    .dtype_format(DataType.U32_Default) \
-    .dtype_format(DataType.U64_Default) \
-    .dtype_format(DataType.F32_Default) \
-    .get_op_info()
-
-
-rrb_destroy_op_info = AiCPURegOp("ReservoirReplayBufferDestroy") \
-    .output(0, "handle", "required") \
-    .attr("handle", "int") \
-    .dtype_format(DataType.I64_Default) \
-    .get_op_info()
-
-
-@op_info_register(rrb_create_op_info)
-def _rrb_create_op_cpu():
-    """ReservoirReplayBufferCreate AICPU register"""
-    return
-
-
-@op_info_register(rrb_push_op_info)
-def _rrb_push_op_cpu():
-    """ReservoirReplayBufferPush AICPU register"""
-    return
-
-
-@op_info_register(rrb_sample_op_info)
-def _rrb_sample_op_cpu():
-    """ReservoirReplayBufferSample AICPU register"""
-    return
-
-
-@op_info_register(rrb_destroy_op_info)
-def _rrb_destroy_op_cpu():
-    """ReservoirReplayBufferDestroy AICPU register"""
-    return

mindspore/ops/_op_impl/aicpu/sparse_cross.py

@@ -1,42 +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.
-# ============================================================================
-
-"""SparseCross op"""
-from mindspore.ops.op_info_register import op_info_register, AiCPURegOp, DataType
-
-sparse_cross_op_info = AiCPURegOp("SparseCross")                                      \
-    .fusion_type("OPAQUE")                                                            \
-    .attr("N", "int")                                                                  \
-    .attr("hashed_output", "bool")                                                    \
-    .attr("hash_key", "int")                                                          \
-    .attr("out_type", "Type")                                                         \
-    .attr("internal_type", "Type")                                                    \
-    .attr("num_buckets", "int")                                                       \
-    .input(0, "indices", "dynamic")                                                   \
-    .input(1, "values", "dynamic")                                                    \
-    .input(2, "shapes", "dynamic")                                                    \
-    .input(3, "dense_inputs", "dynamic")                                              \
-    .output(0, "output_indices", "required")                                          \
-    .output(1, "output_values", "required")                                           \
-    .output(2, "output_shape", "required")                                            \
-    .dtype_format(DataType.I64_Default, DataType.I64_Default, DataType.I64_Default, \
-    DataType.I64_Default, DataType.I64_Default, DataType.I64_Default, DataType.I64_Default) \
-    .get_op_info()
-
-
-@op_info_register(sparse_cross_op_info)
-def _sparse_cross_aicpu():
-    """SparseCross AiCPU register"""
-    return

mindspore/ops/_op_impl/cpu/buffer_get.py

@@ -1,28 +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.
-
-"""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

@@ -1,28 +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.
-
-"""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

mindspore/ops/_op_impl/cpu/priority_replay_buffer.py

@@ -1,42 +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.
-# ============================================================================
-
-"""PriorityReplayBuffer op"""
-from mindspore.ops.op_info_register import op_info_register, CpuRegOp, DataType
-
-prb_sample_op_info = CpuRegOp("PriorityReplayBufferSample") \
-    .input(0, "x", "dynamic") \
-    .output(0, "y", "dynamic") \
-    .dtype_format(DataType.I32_Default, DataType.I32_Default) \
-    .get_op_info()
-
-
-@op_info_register(prb_sample_op_info)
-def _prb_sample_op_cpu():
-    """PriorityReplayBufferSample cpu register"""
-    return
-
-
-prb_push_op_info = CpuRegOp("PriorityReplayBufferPush") \
-    .input(0, "x", "dynamic") \
-    .output(0, "y", "dynamic") \
-    .dtype_format(DataType.I32_Default, DataType.I32_Default) \
-    .get_op_info()
-
-
-@op_info_register(prb_push_op_info)
-def _prb_push_op_cpu():
-    """PriorityReplayBufferPush cpu register"""
-    return