mindspore 2.0.0a0__cp39-cp39-win_amd64.whl → 2.0.0rc1__cp39-cp39-win_amd64.whl

mindspore/ops/op_info_register.py

@@ -23,7 +23,7 @@ import os
 import functools
 
 from mindspore._c_expression import Oplib
-from mindspore._checkparam import Validator as validator
+from mindspore import _checkparam as validator
 
 # path of built-in op info register.
 BUILT_IN_OPS_REGISTER_PATH = "mindspore/ops/_op_impl"
@@ -272,10 +272,12 @@ class RegOp:
             raise ValueError("input size add output size must be equal to dtype format size")
         dtype_format = []
         for arg in args:
-            if not isinstance(arg, tuple) or len(arg) != 2:
-                raise ValueError("dtype and format value must be tuple of two elements")
+            if not isinstance(arg, tuple) or (len(arg) != 2 and len(arg) != 3):
+                raise ValueError("dtype and format value must be tuple of two or three elements")
             self._is_string(arg[0])
             self._is_string(arg[1])
+            if len(arg) == 3:
+                self._is_string(arg[2])
             dtype_format.append(arg)
         self.dtype_format_.append(tuple(dtype_format))
         return self
@@ -291,9 +293,15 @@ class RegOp:
         op_info = {}
         for key, value in self.__dict__.items():
             if isinstance(key, str) and key.endswith('_'):
-                op_info[key.rstrip('_')] = value
-            else:
-                op_info[key] = value
+                key = key.rstrip('_')
+                key_dic = {"dynamic_shape_support": "dynamicShapeSupport",
+                           "dynamic_rank_support": "dynamicRankSupport",
+                           "dynamic_compile_static": "dynamicCompileStatic",
+                           "need_check_support": "needCheckSupport",
+                           "dynamic_format": "dynamicFormat"
+                           }
+                key = key_dic.get(key, key)
+            op_info[key] = value
         return op_info
 
 
@@ -315,7 +323,7 @@ class CpuRegOp(RegOp):
             kwargs (dict): Other information of the input.
         """
         param_list = [index, name, param_type]
-        key_list = ["index", "name", "param_type"]
+        key_list = ["index", "name", "paramType"]
         fn_list = [self._is_int, self._is_string, self._is_string]
         input_dict = self._check_param(param_list, key_list, fn_list, kwargs)
         self.inputs.append(input_dict)
@@ -332,7 +340,7 @@ class CpuRegOp(RegOp):
             kwargs (dict): Other information of the output.
         """
         param_list = [index, name, param_type]
-        key_list = ["index", "name", "param_type"]
+        key_list = ["index", "name", "paramType"]
         fn_list = [self._is_int, self._is_string, self._is_string]
         output_dict = self._check_param(param_list, key_list, fn_list, kwargs)
         self.outputs.append(output_dict)
@@ -375,7 +383,7 @@ class AkgRegOp(RegOp):
             kwargs (dict): Other information of the input.
         """
         param_list = [index, name, param_type]
-        key_list = ["index", "name", "param_type"]
+        key_list = ["index", "name", "paramType"]
         fn_list = [self._is_int, self._is_string, self._is_string]
         input_dict = self._check_param(param_list, key_list, fn_list, kwargs)
         self.inputs.append(input_dict)
@@ -408,7 +416,7 @@ class AkgRegOp(RegOp):
             kwargs (dict): Other information of the attribute.
         """
         param_list = [name, param_type, value_type]
-        key_list = ["name", "param_type", "type"]
+        key_list = ["name", "paramType", "type"]
         fn_list = [self._is_string]
         attr_dict = self._check_param(param_list, key_list, fn_list, kwargs)
         self.attr_.append(attr_dict)
@@ -501,16 +509,16 @@ class TBERegOp(RegOp):
         super(TBERegOp, self).__init__(op_name)
         self.imply_type = "TBE"
         self.async_flag_ = False
-        self.binfile_name_ = ''
+        self.binfile_ = ''
         self.compute_cost_ = 10
-        self.kernel_name_ = ''
+        self.kernel_ = ''
         self.partial_flag_ = False
         self.reshape_type_ = ''
         self.dynamic_rank_support_ = False
-        self.dynamic_shape_ = False
+        self.dynamic_shape_support_ = False
         self.dynamic_compile_static_ = False
-        self.need_check_supported_ = False
-        self.is_dynamic_format_ = False
+        self.need_check_support_ = False
+        self.dynamic_format_ = False
         self.op_pattern_ = ""
         self.real_input_index_ = []
         self.input_to_attr_index_ = []
@@ -584,7 +592,7 @@ class TBERegOp(RegOp):
             binfile_name (str): The binary file name of the operator.
         """
         self._is_string(binfile_name)
-        self.binfile_name_ = binfile_name
+        self.binfile_ = binfile_name
         return self
 
     def compute_cost(self, compute_cost=10):
@@ -607,7 +615,7 @@ class TBERegOp(RegOp):
             kernel_name (str): Name of operator kernel.
         """
         self._is_string(kernel_name)
-        self.kernel_name_ = kernel_name
+        self.kernel_ = kernel_name
         return self
 
     def partial_flag(self, partial_flag=True):
@@ -626,8 +634,9 @@ class TBERegOp(RegOp):
         Reshape type of operator.
 
         Args:
-            reshape_type (str): Value of reshape type. For example, if the input shape is (2,3) and `reshape_type`
-                is set to "CH", then the new shape is (1,2,3,1). "CH" means the C and H dimensions are kept and
+            reshape_type (str): Value of reshape type. For example, if the input shape is :math:`(2, 3)`
+                and `reshape_type` is set to "CH", then the new shape is :math:`(1, 2, 3, 1)`.
+                "CH" means the C and H dimensions are kept and
                 new dimensions are added for N and W dimension.
         """
         self._is_string(reshape_type)
@@ -642,7 +651,7 @@ class TBERegOp(RegOp):
             dynamic_shape (bool): Value of dynamic shape. Default: false.
         """
         self._is_bool(dynamic_shape)
-        self.dynamic_shape_ = dynamic_shape
+        self.dynamic_shape_support_ = dynamic_shape
         return self
 
     def dynamic_compile_static(self, dynamic_compile_static=False):
@@ -664,7 +673,7 @@ class TBERegOp(RegOp):
             need_check_supported (bool): Value of need_check_supported. Default: false.
         """
         self._is_bool(need_check_supported)
-        self.need_check_supported_ = need_check_supported
+        self.need_check_support_ = need_check_supported
         return self
 
     def is_dynamic_format(self, is_dynamic_format=False):
@@ -675,7 +684,7 @@ class TBERegOp(RegOp):
             is_dynamic_format (bool): Value of is_dynamic_format. Default: false.
         """
         self._is_bool(is_dynamic_format)
-        self.is_dynamic_format_ = is_dynamic_format
+        self.dynamic_format_ = is_dynamic_format
         return self
 
     def op_pattern(self, pattern=None):
@@ -702,7 +711,7 @@ class TBERegOp(RegOp):
             kwargs (dict): Other information of the attribute.
         """
         param_list = [name, param_type, value_type, value, default_value]
-        key_list = ["name", "param_type", "type", "value", "default_value"]
+        key_list = ["name", "paramType", "type", "value", "defaultValue"]
         fn_list = [self._is_string]
         attr_dict = self._check_param(param_list, key_list, fn_list, kwargs)
         self.attr_.append(attr_dict)
@@ -722,7 +731,7 @@ class TBERegOp(RegOp):
             kwargs (dict): Other information of the input.
         """
         param_list = [index, name, need_compile, param_type, shape, value_depend]
-        key_list = ["index", "name", "need_compile", "param_type", "shape", "value_depend"]
+        key_list = ["index", "name", "needCompile", "paramType", "shape", "valueDepend"]
         fn_list = [self._is_int, self._is_string, self._is_bool, self._is_string, self._is_string, self._is_string]
         input_dict = self._check_param(param_list, key_list, fn_list, kwargs)
         value_depend_values = ("ignored", "optional", "required")
@@ -745,7 +754,7 @@ class TBERegOp(RegOp):
             kwargs (dict): Other information of the output.
         """
         param_list = [index, name, need_compile, param_type, shape]
-        key_list = ["index", "name", "need_compile", "param_type", "shape"]
+        key_list = ["index", "name", "need_compile", "paramType", "shape"]
         fn_list = [self._is_int, self._is_string, self._is_bool, self._is_string, self._is_string]
         output_dict = self._check_param(param_list, key_list, fn_list, kwargs)
         self.outputs.append(output_dict)
@@ -807,7 +816,7 @@ class CustomRegOp(RegOp):
             TypeError: If `param_type` is neither str nor None.
         """
         param_list = [index, name, param_type]
-        key_list = ["index", "name", "param_type"]
+        key_list = ["index", "name", "paramType"]
         fn_list = [self._is_int, self._is_string, self._is_string]
         input_dict = self._check_param(param_list, key_list, fn_list, kwargs)
         self.inputs.append(input_dict)
@@ -842,7 +851,7 @@ class CustomRegOp(RegOp):
             TypeError: If `param_type` is neither str nor None.
         """
         param_list = [index, name, param_type]
-        key_list = ["index", "name", "param_type"]
+        key_list = ["index", "name", "paramType"]
         fn_list = [self._is_int, self._is_string, self._is_string]
         output_dict = self._check_param(param_list, key_list, fn_list, kwargs)
         self.outputs.append(output_dict)
@@ -917,7 +926,7 @@ class CustomRegOp(RegOp):
             TypeError: If `default_value` is neither str nor None.
         """
         param_list = [name, param_type, value_type, default_value]
-        key_list = ["name", "param_type", "type", "default_value"]
+        key_list = ["name", "paramType", "type", "defaultValue"]
         fn_list = [self._is_string]
         attr_dict = self._check_param(param_list, key_list, fn_list, kwargs)
         self.attr_.append(attr_dict)
@@ -1130,6 +1139,7 @@ class DataType:
 
     None_None = ("", "")
     None_Default = ("", "DefaultFormat")
+
     BOOL_None = ("bool", "")
     BOOL_Default = ("bool", "DefaultFormat")
     BOOL_5HD = ("bool", "NC1HWC0")
@@ -1141,6 +1151,8 @@ class DataType:
     BOOL_HWCN = ("bool", "HWCN")
     BOOL_NDHWC = ("bool", "NDHWC")
     BOOL_ChannelLast = ("bool", "ChannelLast")
+    BOOL_Default_Tuple = ("bool", "DefaultFormat", "tuple")
+    BOOL_Default_List = ("bool", "DefaultFormat", "list")
 
     I8_None = ("int8", "")
     I8_Default = ("int8", "DefaultFormat")
@@ -1152,8 +1164,12 @@ class DataType:
     I8_NHWC = ("int8", "NHWC")
     I8_HWCN = ("int8", "HWCN")
     I8_NDHWC = ("int8", "NDHWC")
+    I8_NCDHW = ("int8", "NCDHW")
     I8_ChannelLast = ("int8", "ChannelLast")
     I8_NDC1HWC0 = ("int8", "NDC1HWC0")
+    I8_NC1HWC0 = ("int8", "NC1HWC0")
+    I8_Default_Tuple = ("int8", "DefaultFormat", "tuple")
+    I8_Default_List = ("int8", "DefaultFormat", "list")
 
     U8_None = ("uint8", "")
     U8_Default = ("uint8", "DefaultFormat")
@@ -1165,8 +1181,12 @@ class DataType:
     U8_NHWC = ("uint8", "NHWC")
     U8_HWCN = ("uint8", "HWCN")
     U8_NDHWC = ("uint8", "NDHWC")
+    U8_NCDHW = ("uint8", "NCDHW")
     U8_ChannelLast = ("uint8", "ChannelLast")
     U8_NDC1HWC0 = ("uint8", "NDC1HWC0")
+    U8_NC1HWC0 = ("uint8", "NC1HWC0")
+    U8_Default_Tuple = ("uint8", "DefaultFormat", "tuple")
+    U8_Default_List = ("uint8", "DefaultFormat", "list")
 
     I16_None = ("int16", "")
     I16_Default = ("int16", "DefaultFormat")
@@ -1179,6 +1199,8 @@ class DataType:
     I16_HWCN = ("int16", "HWCN")
     I16_NDHWC = ("int16", "NDHWC")
     I16_ChannelLast = ("int16", "ChannelLast")
+    I16_Default_Tuple = ("int16", "DefaultFormat", "tuple")
+    I16_Default_List = ("int16", "DefaultFormat", "list")
 
     U16_None = ("uint16", "")
     U16_Default = ("uint16", "DefaultFormat")
@@ -1191,6 +1213,8 @@ class DataType:
     U16_HWCN = ("uint16", "HWCN")
     U16_NDHWC = ("uint16", "NDHWC")
     U16_ChannelLast = ("uint16", "ChannelLast")
+    U16_Default_Tuple = ("uint16", "DefaultFormat", "tuple")
+    U16_Default_List = ("uint16", "DefaultFormat", "list")
 
     I32_None = ("int32", "")
     I32_Default = ("int32", "DefaultFormat")
@@ -1202,7 +1226,11 @@ class DataType:
     I32_NHWC = ("int32", "NHWC")
     I32_HWCN = ("int32", "HWCN")
     I32_NDHWC = ("int32", "NDHWC")
+    I32_NDC1HWC0 = ("int32", "NDC1HWC0")
+    I32_NCDHW = ("int32", "NCDHW")
     I32_ChannelLast = ("int32", "ChannelLast")
+    I32_Default_Tuple = ("int32", "DefaultFormat", "tuple")
+    I32_Default_List = ("int32", "DefaultFormat", "list")
 
     U32_None = ("uint32", "")
     U32_Default = ("uint32", "DefaultFormat")
@@ -1215,6 +1243,8 @@ class DataType:
     U32_HWCN = ("uint32", "HWCN")
     U32_NDHWC = ("uint32", "NDHWC")
     U32_ChannelLast = ("uint32", "ChannelLast")
+    U32_Default_Tuple = ("uint32", "DefaultFormat", "tuple")
+    U32_Default_List = ("uint32", "DefaultFormat", "list")
 
     I64_None = ("int64", "")
     I64_Default = ("int64", "DefaultFormat")
@@ -1227,6 +1257,8 @@ class DataType:
     I64_HWCN = ("int64", "HWCN")
     I64_NDHWC = ("int64", "NDHWC")
     I64_ChannelLast = ("int64", "ChannelLast")
+    I64_Default_Tuple = ("int64", "DefaultFormat", "tuple")
+    I64_Default_List = ("int64", "DefaultFormat", "list")
 
     U64_None = ("uint64", "")
     U64_Default = ("uint64", "DefaultFormat")
@@ -1239,6 +1271,8 @@ class DataType:
     U64_HWCN = ("uint64", "HWCN")
     U64_NDHWC = ("uint64", "NDHWC")
     U64_ChannelLast = ("uint64", "ChannelLast")
+    U64_Default_Tuple = ("uint64", "DefaultFormat", "tuple")
+    U64_Default_List = ("uint64", "DefaultFormat", "list")
 
     F16_None = ("float16", "")
     F16_Default = ("float16", "DefaultFormat")
@@ -1258,6 +1292,8 @@ class DataType:
     F16_FracZNRNN = ("float16", "FRACTAL_ZN_RNN")
     F16_ND_RNNBIAS = ("float16", "ND_RNN_BIAS")
     F16_ChannelLast = ("float16", "ChannelLast")
+    F16_Default_Tuple = ("float16", "DefaultFormat", "tuple")
+    F16_Default_List = ("float16", "DefaultFormat", "list")
 
     F32_None = ("float32", "")
     F32_Default = ("float32", "DefaultFormat")
@@ -1277,6 +1313,8 @@ class DataType:
     F32_FracZNRNN = ("float32", "FRACTAL_ZN_RNN")
     F32_ND_RNNBIAS = ("float32", "ND_RNN_BIAS")
     F32_ChannelLast = ("float32", "ChannelLast")
+    F32_Default_Tuple = ("float32", "DefaultFormat", "tuple")
+    F32_Default_List = ("float32", "DefaultFormat", "list")
 
     F64_None = ("float64", "")
     F64_Default = ("float64", "DefaultFormat")
@@ -1289,6 +1327,10 @@ class DataType:
     F64_HWCN = ("float64", "HWCN")
     F64_NDHWC = ("float64", "NDHWC")
     F64_ChannelLast = ("float64", "ChannelLast")
+    F64_Default_Tuple = ("float64", "DefaultFormat", "tuple")
+    F64_Default_List = ("float64", "DefaultFormat", "list")
 
     C64_Default = ("complex64", "DefaultFormat")
     C128_Default = ("complex128", "DefaultFormat")
+    C64_Default_Tuple = ("complex64", "DefaultFormat", "tuple")
+    C128_Default_Tuple = ("complex128", "DefaultFormat", "tuple")

mindspore/ops/operations/__init__.py

@@ -21,7 +21,7 @@ A collection of operators to build neural networks or to compute functions.
 
 from ._embedding_cache_ops import (CacheSwapTable, UpdateCache, MapCacheIdx, SubAndFilter,
                                    MapUniform, DynamicAssign, PadAndShift)
-from ._inner_ops import (FillV2, MatmulDDS, DSDMatmul, Cummin, ExtractImagePatches)
+from ._inner_ops import (MatmulDDS, DSDMatmul, Cummin, ExtractImagePatches)
 from ._quant_ops import *
 from ._thor_ops import (CusBatchMatMul, CusCholeskyTrsm, CusFusedAbsMax1, CusImg2Col, CusMatMulCubeDenseLeft,
                         CusMatMulCubeFraczRightMul, CusMatMulCube, CusMatrixCombine, CusTranspose02314,
@@ -35,7 +35,7 @@ from .array_ops import (ArgMaxWithValue, ArgMinWithValue, Argmax, Argmin, BatchT
                         Eye, Fill, Gather, GatherD, GatherNd, GatherV2, Identity, Im2Col, InvertPermutation,
                         LowerBound, Lstsq, MaskedFill, MaskedSelect, Meshgrid, Mvlgamma, Ones, OnesLike,
                         Pack, Padding, ParallelConcat, PopulationCount, Range, Rank, Reshape, ResizeNearestNeighbor,
-                        ReverseSequence, ReverseV2, Rint, ScalarToTensor, ScatterAdd,
+                        ReverseSequence, ReverseV2, Rint, ScalarToArray, ScalarToTensor, ScatterAdd,
                         ScatterDiv, ScatterMax, ScatterMin, ScatterMul, ScatterNd, ScatterNdAdd, ScatterNdDiv,
                         ScatterNdMax, ScatterNdMin, ScatterNdSub, ScatterNdUpdate, ScatterNonAliasingAdd, ScatterSub,
                         ScatterUpdate, SearchSorted, Select, Shape, Size, Slice, Sort, SpaceToBatch, SpaceToBatchND,
@@ -48,7 +48,7 @@ from .array_ops import (ArgMaxWithValue, ArgMinWithValue, Argmax, Argmin, BatchT
                         MatrixDiagPartV3, MatrixDiagV3, MatrixSetDiagV3, NonZero, Expand, Col2Im, ConjugateTranspose,
                         FillDiagonal, Fills, ResizeNearestNeighborV2, RightShift, ScatterAddWithAxis,
                         ScatterNdMul, SegmentMean, SegmentProd, SegmentSum, SegmentMax, SegmentMin, Tril, Triu,
-                        UniqueConsecutive, UnravelIndex)
+                        UniqueConsecutive, UnravelIndex, FillV2, CountNonZero, TensorScatterElements)
 from .comm_ops import (AllGather, AllReduce, NeighborExchange, NeighborExchangeV2, AlltoAll, _AllSwap, ReduceScatter,
                        Broadcast,
                        _MirrorOperator, _MirrorMiniStepOperator, _MiniStepAllGather, ReduceOp, _VirtualDataset,
@@ -61,14 +61,14 @@ from .debug_ops import (ImageSummary, InsertGradientOf, HookBackward, ScalarSumm
                         TensorSummary, HistogramSummary, Print, Assert)
 from .image_ops import (CropAndResize, NonMaxSuppressionV3, HSVToRGB, AdjustHue, AdjustSaturation,
                         NonMaxSuppressionWithOverlaps, ResizeArea, ResizeBilinearV2, ExtractGlimpse,
-                        CombinedNonMaxSuppression, RGBToHSV, ScaleAndTranslate)
+                        CombinedNonMaxSuppression, RGBToHSV, ScaleAndTranslate, ResizeLinear1D, ResizeBicubic)
 from .inner_ops import (ScalarCast, Randperm, NoRepeatNGram, LambApplyOptimizerAssign, LambApplyWeightAssign,
                         FusedWeightScaleApplyMomentum, FusedCastAdamWeightDecay, FusedAdaFactor,
                         FusedAdaFactorWithGlobalNorm)
 from .linalg_ops import (Svd, Geqrf)
 from .math_ops import (Abs, ACos, Asin, Asinh, AddN, AccumulateNV2, AssignAdd, AssignSub, Atan2, BatchMatMul,
-                       BitwiseAnd, BitwiseOr, Ger,
-                       BitwiseXor, Inv, Invert, ApproximateEqual, InplaceAdd, InplaceSub, InplaceUpdate,
+                       BitwiseAnd, BitwiseOr, Ger, BitwiseXor, Inv, Invert, ApproximateEqual,
+                       InplaceAdd, InplaceSub, InplaceUpdate, InplaceUpdateV2,
                        ReduceMax, ReduceMin, ReduceMean, ReduceSum, ReduceAll, ReduceProd, CumProd, Cdist, ReduceAny,
                        Cos, Cross, Div, DivNoNan, Equal, EqualCount, Exp, Expm1, Erf, Erfc, Floor, FloorDiv, FloorMod,
                        Ceil, Acosh, Greater, GreaterEqual, Lerp, Less, LessEqual, Log, Log1p, LogicalAnd, Mod,
@@ -85,8 +85,9 @@ from .math_ops import (Abs, ACos, Asin, Asinh, AddN, AccumulateNV2, AssignAdd, A
                        FFTWithSize, Heaviside, Histogram, Hypot, Lcm, LuUnpack, MatrixExp,
                        MatrixLogarithm, MatrixPower, MatrixSolve, MatrixTriangularSolve, ReduceStd, STFT,
                        NextAfter, Orgqr, Qr, RaggedRange, Digamma, Eig, EuclideanNorm, CompareAndBitpack, ComplexAbs,
-                       CumulativeLogsumexp, Gcd, Trace, TridiagonalMatMul, TrilIndices, TriuIndices, Zeta,
-                       Roll)
+                       CumulativeLogsumexp, Gcd, Polygamma, Trace, TridiagonalMatMul, TrilIndices, TriuIndices, Zeta,
+                       Roll, Lgamma, Logit, MatrixSolveLs, Polar, Fmax, Fmin, Quantile, Sinc, Angle, Bernoulli,
+                       NanToNum, Igamma, BesselI0, BesselI1, InplaceIndexAdd, Igammac)
 from .nn_ops import (LSTM, SGD, Adam, AdamWeightDecay, FusedSparseAdam, FusedSparseLazyAdam, AdamNoUpdateParam,
                      ApplyMomentum, BatchNorm, BiasAdd, Conv2D, Conv3D, Conv2DTranspose, Conv3DTranspose,
                      DepthwiseConv2dNative,
@@ -97,7 +98,8 @@ from .nn_ops import (LSTM, SGD, Adam, AdamWeightDecay, FusedSparseAdam, FusedSpa
                      LogSoftmax, MaxPool3D, AvgPool3D,
                      MaxPool, DataFormatDimMap,
                      AvgPool, Conv2DBackpropInput, ComputeAccidentalHits,
-                     MaxPoolWithArgmax, OneHot, Pad, MirrorPad, Mish, PReLU, ReLU, ReLU6, ReLUV2, HSwish, HSigmoid,
+                     MaxPoolWithArgmax, MaxPoolWithArgmaxV2, OneHot, Pad, MirrorPad, Mish, PReLU, ReLU, ReLU6, ReLUV2,
+                     HSwish, HSigmoid,
                      ResizeBilinear, Sigmoid, SeLU, HShrink, ApplyKerasMomentum,
                      SigmoidCrossEntropyWithLogits, NLLLoss, BCEWithLogitsLoss,
                      SmoothL1Loss, SoftMarginLoss, Softmax, Softsign, Softplus, LRN, RNNTLoss, DynamicRNN, DynamicGRUV2,
@@ -111,9 +113,10 @@ from .nn_ops import (LSTM, SGD, Adam, AdamWeightDecay, FusedSparseAdam, FusedSpa
                      ApplyRMSProp, ApplyCenteredRMSProp, BasicLSTMCell, InTopK, AdaptiveAvgPool2D, SoftShrink,
                      ApplyAdamWithAmsgrad, AdaptiveAvgPool3D, AdaptiveMaxPool2D, AdaptiveMaxPool3D,
                      GridSampler3D, MaxPool3DWithArgmax, MaxUnpool2D, NuclearNorm, NthElement, MultilabelMarginLoss,
-                     PSROIPooling, Dilation2D, DataFormatVecPermute, DeformableOffsets, FractionalAvgPool,
+                     Dilation2D, DataFormatVecPermute, DeformableOffsets, FractionalAvgPool,
                      FractionalMaxPool, FractionalMaxPool3DWithFixedKsize, FractionalMaxPoolWithFixedKsize,
-                     GridSampler2D, TripletMarginLoss, UpsampleNearest3D, UpsampleTrilinear3D, PadV3)
+                     GridSampler2D, TripletMarginLoss, UpsampleNearest3D, UpsampleTrilinear3D, PadV3, ChannelShuffle,
+                     GLU, MaxUnpool3D, Pdist)
 from .other_ops import (Assign, IOU, BoundingBoxDecode, BoundingBoxEncode,
                         ConfusionMatrix, UpdateState, Load, StopGradient,
                         CheckValid, Partial, Depend, identity, Push, Pull, PyExecute, PyFunc, _DynamicLossScale,
@@ -121,9 +124,9 @@ from .other_ops import (Assign, IOU, BoundingBoxDecode, BoundingBoxEncode,
 from .random_ops import (RandomChoiceWithMask, StandardNormal, Gamma, RandomGamma, Poisson, UniformInt, UniformReal,
                          RandomCategorical, StandardLaplace, Multinomial, UniformCandidateSampler,
                          LogUniformCandidateSampler, TruncatedNormal, LogNormalReverse, NonDeterministicInts,
-                         ParameterizedTruncatedNormal, RandomPoisson)
+                         ParameterizedTruncatedNormal, RandomPoisson, MultinomialWithReplacement, RandomShuffle)
 from .rl_ops import (BufferAppend, BufferGetItem, BufferSample)
-from .sparse_ops import (SparseToDense, SparseTensorDenseMatmul, SparseTensorDenseAdd)
+from .sparse_ops import (SparseToDense, SparseTensorDenseMatmul, SparseTensorDenseAdd, SparseSlice)
 from .spectral_ops import (BartlettWindow, BlackmanWindow)
 
 __all__ = [
@@ -173,6 +176,7 @@ __all__ = [
     'FillV2',
     'Flatten',
     'MaxPoolWithArgmax',
+    'MaxPoolWithArgmaxV2',
     'BNTrainingReduce',
     'BNTrainingUpdate',
     'BatchNorm',
@@ -321,6 +325,7 @@ __all__ = [
     'DynamicRNN',
     'ReduceAll',
     'ReduceAny',
+    'ScalarToArray',
     'ScalarToTensor',
     'TupleToArray',
     'GeSwitch',
@@ -470,6 +475,7 @@ __all__ = [
     "DataFormatDimMap",
     "ApproximateEqual",
     "InplaceUpdate",
+    "InplaceUpdateV2",
     "InTopK",
     "UniformCandidateSampler",
     "LogUniformCandidateSampler",
@@ -583,7 +589,6 @@ __all__ = [
     "Orgqr",
     "NonDeterministicInts",
     "NonZero",
-    "PSROIPooling",
     "ResizeArea",
     "ResizeBilinearV2",
     "Qr",
@@ -643,6 +648,34 @@ __all__ = [
     "Zeta",
     "PadV3",
     "Roll",
+    "Lgamma",
+    "SparseSlice",
+    "ResizeLinear1D",
+    "ResizeBicubic",
+    "Logit",
+    "MatrixSolveLs",
+    "Polar",
+    "Fmax",
+    "Fmin",
+    "Polygamma",
+    "CountNonZero",
+    "Quantile",
+    "Sinc",
+    "Angle",
+    "Bernoulli",
+    "NanToNum",
+    "ChannelShuffle",
+    "Igamma",
+    "BesselI0",
+    "MultinomialWithReplacement",
+    "BesselI1",
+    "GLU",
+    "TensorScatterElements",
+    "RandomShuffle",
+    "InplaceIndexAdd",
+    "Igammac",
+    "MaxUnpool3D",
+    "Pdist"
 ]
 
 __custom__ = [

mindspore/ops/operations/_csr_ops.py

@@ -22,7 +22,7 @@ class CSRReduceSum(Primitive):
     in the dimension.
 
     .. warning::
-        This is an experimental prototype that is subject to change and/or deletion.
+        This is an experimental API that is subject to change or deletion.
 
     Inputs:
         - **indptr** (Tensor) - A Tensor.
@@ -72,7 +72,7 @@ class CSRMV(Primitive):
     Sparse matrix-vector multiplication.
 
     .. warning::
-        This is an experimental prototype that is subject to change and/or deletion.
+        This is an experimental API that is subject to change or deletion.
 
     Inputs:
         - **indptr** (Tensor) - A Tensor.
@@ -123,7 +123,7 @@ class CSRMul(Primitive):
     Elemwise multiplication of a CSRTensor (composed of `indptr`, `indices`, `values` and `shape`) and a dense tensor.
 
     .. warning::
-        This is an experimental prototype that is subject to change and/or deletion.
+        This is an experimental API that is subject to change or deletion.
 
     Note:
         The op outputs a 1-D dense tensor whose shape and values are the same as input `values`.
@@ -177,7 +177,7 @@ class CSRGather(Primitive):
     Returns the values of a CSRTensor indexed from a dense tensor using indptr and indices.
 
     .. warning::
-        This is an experimental prototype that is subject to change and/or deletion.
+        This is an experimental API that is subject to change or deletion.
 
     Inputs:
         - **indptr** (Tensor) - A Tensor.
@@ -225,7 +225,7 @@ class CSR2COO(Primitive):
     Converts the indptr of a CSRTensor to the row indices of a COOTensor.
 
     .. warning::
-        This is an experimental prototype that is subject to change and/or deletion.
+        This is an experimental API that is subject to change or deletion.
 
     Inputs:
         - **indptr** (Tensor) - A Tensor.
@@ -265,7 +265,7 @@ class COO2CSR(Primitive):
     Converts the row indices of a COOTensor to the indptr of a CSRTensor.
 
     .. warning::
-        This is an experimental prototype that is subject to change and/or deletion.
+        This is an experimental API that is subject to change or deletion.
 
     Inputs:
         - **row_indices** (Tensor) - A Tensor.
@@ -357,7 +357,7 @@ class CSRMM(Primitive):
     Sparse matrix-vector multiplication.
 
     .. warning::
-        This is an experimental prototype that is subject to change and/or deletion.
+        This is an experimental API that is subject to change or deletion.
 
     Inputs:
         - **indptr** (Tensor) - A Tensor.

mindspore/ops/operations/_embedding_cache_ops.py

@@ -15,7 +15,7 @@
 """cache_ops"""
 
 from __future__ import absolute_import
-from mindspore._checkparam import Validator as validator
+from mindspore import _checkparam as validator
 from mindspore.common import dtype as mstype
 from mindspore.ops.primitive import prim_attr_register, PrimitiveWithCheck
 from mindspore.ops import signature as sig
@@ -269,10 +269,10 @@ class PadAndShift(PrimitiveWithCheck):
 
     Inputs:
         - **input_x** (Tensor) - The input Tensor, which will be copied
-            to `output`.
+          to `output`.
         - **cum_sum_arr** (Tensor) - The last value of cum_sum_arr is
-            the pad length of output tensor, `cum_sum_arr[shift_idx]` is
-            the start to shift, and `cum_sum_arr[shift_idx+1]` is the end.
+          the pad length of output tensor, `cum_sum_arr[shift_idx]` is
+          the start to shift, and `cum_sum_arr[shift_idx+1]` is the end.
         - **shift_idx** (int) - The idx of `cum_sum_arr` .
 
     Outputs:
@@ -281,7 +281,7 @@ class PadAndShift(PrimitiveWithCheck):
     Raises:
         TypeError: `input_x` or `cum_sum_arr` is not Tensor.
         TypeError: `shift_idx` is not int.
-        ValueError: Value of `shift_idx` is larger than or equal to the length of  `cum_sum_arr` .
+        ValueError: Value of `shift_idx` is larger than or equal to the length of `cum_sum_arr` .
 
     Supported Platforms:
         `CPU`