mindspore 1.10.0__cp38-cp38-win_amd64.whl → 2.0.0rc1__cp38-cp38-win_amd64.whl

mindspore/ops/operations/__init__.py

@@ -21,29 +21,34 @@ 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)
+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,
                         CusMatMulCubeDenseRight, CusMatMulCubeFraczLeftCast, NewIm2Col,
-                        LoadIm2Col, UpdateThorGradient, Cholesky, CholeskyTrsm,
+                        LoadIm2Col, UpdateThorGradient, CholeskyTrsm,
                         DetTriangle, ProdForceSeA)
-from ._ms_kernel import (ms_kernel, ms_hybrid)
-from .array_ops import (ArgMaxWithValue, ArgMinWithValue, Argmax, Argmin, BatchToSpace, BatchToSpaceND, BroadcastTo,
-                        Cast, Coalesce, Concat, Cummax, DType, DepthToSpace, Diag, DiagPart, DynamicShape, EditDistance,
-                        EmbeddingLookup, ExpandDims, ExtractVolumePatches, Eye, Fill, Gather, GatherD, GatherNd,
-                        GatherV2, Identity, Im2Col, InvertPermutation, IsInstance, IsSubClass, LowerBound, Lstsq,
-                        MaskedFill, MaskedSelect, Meshgrid, Mvlgamma, Ones, OnesLike, Pack, Padding, ParallelConcat,
-                        PopulationCount, Range, Rank, Reshape, ResizeNearestNeighbor, ReverseSequence, ReverseV2, Rint,
-                        SameTypeShape, 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, SpaceToDepth, SparseGatherV2, Split, SplitV,
-                        Squeeze, Stack, StridedSlice, TensorScatterAdd, TensorScatterDiv, TensorScatterMax,
-                        TensorScatterMin, TensorScatterMul, TensorScatterSub, TensorScatterUpdate, TensorShape, Tile,
-                        TopK, TransShape, Transpose, TupleToArray, Unique, UniqueWithPad, Unpack, UnsortedSegmentMax,
-                        UnsortedSegmentMin, UnsortedSegmentProd, UnsortedSegmentSum, Unstack, UpperBound, Zeros,
-                        ZerosLike)
+from ._ms_kernel import (ms_kernel, kernel)
+from .array_ops import (ArgMaxWithValue, ArgMinWithValue, Argmax, Argmin, BatchToSpace, BatchToSpaceND,
+                        BatchToSpaceNDV2, BroadcastTo, Cast, Coalesce, Concat, Cummax, DType, DepthToSpace, Diag,
+                        DiagPart, DynamicShape, EditDistance, EmbeddingLookup, ExpandDims, ExtractVolumePatches,
+                        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, 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,
+                        SpaceToDepth, SparseGatherV2, Split, SplitV, Squeeze, Stack, StridedSlice, TensorScatterAdd,
+                        TensorScatterDiv, TensorScatterMax, TensorScatterMin, TensorScatterMul, TensorScatterSub,
+                        TensorScatterUpdate, TensorShape, Tile, TopK, TransShape, Transpose, TupleToArray, Unique,
+                        UniqueWithPad, Unpack, UnsortedSegmentMax, UnsortedSegmentMin, UnsortedSegmentProd,
+                        UnsortedSegmentSum, Unstack, UpperBound, Zeros, ZerosLike, AffineGrid, Bincount, CheckNumerics,
+                        HammingWindow, IdentityN, IndexFill, LeftShift, ListDiff, LogSpace, MatrixBandPart,
+                        MatrixDiagPartV3, MatrixDiagV3, MatrixSetDiagV3, NonZero, Expand, Col2Im, ConjugateTranspose,
+                        FillDiagonal, Fills, ResizeNearestNeighborV2, RightShift, ScatterAddWithAxis,
+                        ScatterNdMul, SegmentMean, SegmentProd, SegmentSum, SegmentMax, SegmentMin, Tril, Triu,
+                        UniqueConsecutive, UnravelIndex, FillV2, CountNonZero, TensorScatterElements)
 from .comm_ops import (AllGather, AllReduce, NeighborExchange, NeighborExchangeV2, AlltoAll, _AllSwap, ReduceScatter,
                        Broadcast,
                        _MirrorOperator, _MirrorMiniStepOperator, _MiniStepAllGather, ReduceOp, _VirtualDataset,
@@ -54,13 +59,16 @@ from .control_ops import GeSwitch, Merge
 from .custom_ops import (Custom)
 from .debug_ops import (ImageSummary, InsertGradientOf, HookBackward, ScalarSummary,
                         TensorSummary, HistogramSummary, Print, Assert)
-from .image_ops import (CropAndResize, NonMaxSuppressionV3, HSVToRGB)
+from .image_ops import (CropAndResize, NonMaxSuppressionV3, HSVToRGB, AdjustHue, AdjustSaturation,
+                        NonMaxSuppressionWithOverlaps, ResizeArea, ResizeBilinearV2, ExtractGlimpse,
+                        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,
@@ -72,7 +80,14 @@ from .math_ops import (Abs, ACos, Asin, Asinh, AddN, AccumulateNV2, AssignAdd, A
                        Sin, Sqrt, Rsqrt, BesselI0e, BesselI1e, TruncateDiv, TruncateMod, Addcdiv,
                        Addcmul, Square, Sub, TensorAdd, Add, Sign, Round, SquareSumAll, Atan, Atanh, Cosh, Sinh, Eps,
                        Tan, MatrixInverse, IndexAdd, Erfinv, Conj, Real, Imag, Complex, Trunc, IsClose, LuSolve,
-                       CholeskyInverse)
+                       CholeskyInverse, BesselJ0, BesselJ1, BesselK0, BesselK0e, BesselK1, BesselK1e, BesselY0,
+                       BesselY1, Bucketize, Cauchy, Cholesky, CholeskySolve, Betainc,
+                       FFTWithSize, Heaviside, Histogram, Hypot, Lcm, LuUnpack, MatrixExp,
+                       MatrixLogarithm, MatrixPower, MatrixSolve, MatrixTriangularSolve, ReduceStd, STFT,
+                       NextAfter, Orgqr, Qr, RaggedRange, Digamma, Eig, EuclideanNorm, CompareAndBitpack, ComplexAbs,
+                       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,
@@ -83,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,
@@ -95,32 +111,23 @@ from .nn_ops import (LSTM, SGD, Adam, AdamWeightDecay, FusedSparseAdam, FusedSpa
                      ApplyAdaMax, ApplyAdadelta, ApplyAdagrad, ApplyAdagradV2, MultiMarginLoss, ApplyAdagradDA,
                      ApplyAddSign, ApplyPowerSign, ApplyGradientDescent, ApplyProximalGradientDescent,
                      ApplyRMSProp, ApplyCenteredRMSProp, BasicLSTMCell, InTopK, AdaptiveAvgPool2D, SoftShrink,
-                     ApplyAdamWithAmsgrad)
-from .other_ops import (Assign, IOU, BartlettWindow, BlackmanWindow, BoundingBoxDecode, BoundingBoxEncode,
-                        ConfusionMatrix, UpdateState, Load,
-                        CheckValid, Partial, Depend, identity, CheckBprop, Push, Pull, PyFunc, _DynamicLossScale)
+                     ApplyAdamWithAmsgrad, AdaptiveAvgPool3D, AdaptiveMaxPool2D, AdaptiveMaxPool3D,
+                     GridSampler3D, MaxPool3DWithArgmax, MaxUnpool2D, NuclearNorm, NthElement, MultilabelMarginLoss,
+                     Dilation2D, DataFormatVecPermute, DeformableOffsets, FractionalAvgPool,
+                     FractionalMaxPool, FractionalMaxPool3DWithFixedKsize, FractionalMaxPoolWithFixedKsize,
+                     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,
+                        SampleDistortedBoundingBoxV2)
 from .random_ops import (RandomChoiceWithMask, StandardNormal, Gamma, RandomGamma, Poisson, UniformInt, UniformReal,
                          RandomCategorical, StandardLaplace, Multinomial, UniformCandidateSampler,
-                         LogUniformCandidateSampler)
+                         LogUniformCandidateSampler, TruncatedNormal, LogNormalReverse, NonDeterministicInts,
+                         ParameterizedTruncatedNormal, RandomPoisson, MultinomialWithReplacement, RandomShuffle)
 from .rl_ops import (BufferAppend, BufferGetItem, BufferSample)
-from .sparse_ops import (SparseToDense, SparseTensorDenseMatmul)
-from .sponge_ops import (BondForce, BondEnergy, BondAtomEnergy, BondForceWithAtomEnergy, BondForceWithAtomVirial,
-                         DihedralForce, DihedralEnergy, DihedralAtomEnergy, DihedralForceWithAtomEnergy, AngleForce,
-                         AngleEnergy, AngleAtomEnergy, AngleForceWithAtomEnergy, PMEReciprocalForce,
-                         LJForce, LJEnergy, LJForceWithPMEDirectForce, PMEExcludedForce, PMEEnergy, Dihedral14LJForce,
-                         Dihedral14LJForceWithDirectCF, Dihedral14LJEnergy, Dihedral14LJCFForceWithAtomEnergy,
-                         Dihedral14LJAtomEnergy, Dihedral14CFEnergy, Dihedral14CFAtomEnergy,
-                         MDTemperature, MDIterationLeapFrogLiujian,
-                         CrdToUintCrd, MDIterationSetupRandState, TransferCrd, FFT3D, IFFT3D, NeighborListUpdate)
-from .sponge_update_ops import (ConstrainForceCycleWithVirial, RefreshUintCrd, LastCrdToDr, RefreshCrdVel,
-                                CalculateNowrapCrd, RefreshBoxmapTimes, Totalc6get, CrdToUintCrdQuarter,
-                                MDIterationLeapFrogLiujianWithMaxVel, GetCenterOfMass, MapCenterOfMass,
-                                NeighborListRefresh, MDIterationLeapFrog, MDIterationLeapFrogWithMaxVel,
-                                MDIterationGradientDescent, BondForceWithAtomEnergyAndVirial, ConstrainForceCycle,
-                                LJForceWithVirialEnergy, LJForceWithPMEDirectForceUpdate, PMEReciprocalForceUpdate,
-                                PMEExcludedForceUpdate, LJForceWithVirialEnergyUpdate,
-                                Dihedral14ForceWithAtomEnergyVirial, PMEEnergyUpdate,
-                                ConstrainForceVirial, ConstrainForce, Constrain)
+from .sparse_ops import (SparseToDense, SparseTensorDenseMatmul, SparseTensorDenseAdd, SparseSlice)
+from .spectral_ops import (BartlettWindow, BlackmanWindow)
 
 __all__ = [
     'HSVToRGB',
@@ -169,6 +176,7 @@ __all__ = [
     'FillV2',
     'Flatten',
     'MaxPoolWithArgmax',
+    'MaxPoolWithArgmaxV2',
     'BNTrainingReduce',
     'BNTrainingUpdate',
     'BatchNorm',
@@ -200,8 +208,6 @@ __all__ = [
     'Einsum',
     'Renorm',
     'Cast',
-    'IsSubClass',
-    'IsInstance',
     'Reshape',
     'Squeeze',
     'Transpose',
@@ -269,6 +275,7 @@ __all__ = [
     'RandomChoiceWithMask',
     'StandardNormal',
     'Multinomial',
+    'TruncatedNormal',
     'Gamma',
     'RandomGamma',
     'Mvlgamma',
@@ -323,8 +330,6 @@ __all__ = [
     'TupleToArray',
     'GeSwitch',
     'Merge',
-    'SameTypeShape',
-    'CheckBprop',
     'CheckValid',
     'BartlettWindow',
     'BlackmanWindow',
@@ -359,6 +364,7 @@ __all__ = [
     'Partial',
     'Depend',
     'UpdateState',
+    'StopGradient',
     'identity',
     'AvgPool',
     # Back Primitive
@@ -454,6 +460,7 @@ __all__ = [
     "ApplyCenteredRMSProp",
     "SpaceToBatchND",
     "BatchToSpaceND",
+    "BatchToSpaceNDV2",
     "SquareSumAll",
     "BitwiseAnd",
     "BitwiseOr",
@@ -468,6 +475,7 @@ __all__ = [
     "DataFormatDimMap",
     "ApproximateEqual",
     "InplaceUpdate",
+    "InplaceUpdateV2",
     "InTopK",
     "UniformCandidateSampler",
     "LogUniformCandidateSampler",
@@ -480,6 +488,7 @@ __all__ = [
     "Pull",
     "ReLUV2",
     "SparseToDense",
+    "SparseTensorDenseAdd",
     "SparseTensorDenseMatmul",
     "SparseApplyAdadelta",
     "MatrixInverse",
@@ -497,9 +506,8 @@ __all__ = [
     "TensorScatterMul",
     "TensorScatterDiv",
     "SoftShrink",
-    "FFT3D",
-    "IFFT3D",
     "HShrink",
+    "PyExecute",
     "PyFunc",
     "BufferAppend",
     "BufferGetItem",
@@ -518,79 +526,163 @@ __all__ = [
     "LuSolve",
     "CholeskyInverse",
     "Cummax",
-]
-
-__sponge__ = [
+    "AdjustHue",
+    "AdjustSaturation",
+    "AffineGrid",
+    "BesselJ0",
+    "BesselJ1",
+    "BesselK0",
+    "BesselK0e",
+    "BesselK1",
+    "BesselK1e",
+    "BesselY0",
+    "BesselY1",
+    "Bincount",
+    "Bucketize",
+    "Cauchy",
+    "CheckNumerics",
+    "AdaptiveAvgPool3D",
+    "AdaptiveMaxPool2D",
+    "AdaptiveMaxPool3D",
+    "Cholesky",
+    "CholeskySolve",
+    "CTCLossV2",
+    "Betainc",
+    "FFTWithSize",
+    "GridSampler3D",
+    "HammingWindow",
+    "Heaviside",
+    "Histogram",
+    "Hypot",
+    "IdentityN",
+    "IndexFill",
+    "IsClose",
+    "Lcm",
+    "LeftShift",
+    "ListDiff",
+    "LogNormalReverse",
+    "LogSpace",
+    "LuUnpack",
+    "Im2Col",
+    "MapCacheIdx",
+    "MatrixBandPart",
+    "MatrixDiagPartV3",
+    "MatrixDiagV3",
+    "MatrixExp",
+    "MatrixLogarithm",
+    "MatrixPower",
+    "MatrixSetDiagV3",
+    "MatrixSolve",
+    "MatrixTriangularSolve",
+    "MaxPool3DWithArgmax",
+    "MapUniform",
+    "MaxUnpool2D",
+    "ReduceStd",
+    "STFT",
+    "SubAndFilter",
+    "Svd",
+    "NuclearNorm",
+    "NthElement",
+    "MultilabelMarginLoss",
+    "NonMaxSuppressionWithOverlaps",
+    "NextAfter",
+    "Orgqr",
+    "NonDeterministicInts",
+    "NonZero",
+    "ResizeArea",
+    "ResizeBilinearV2",
+    "Qr",
+    "ParameterizedTruncatedNormal",
+    "RandomGamma",
+    "RandomPoisson",
+    "RaggedRange",
+    "Digamma",
+    "Dilation2D",
+    "Eig",
+    "Expand",
+    "ExtractGlimpse",
+    "EuclideanNorm",
+    "CombinedNonMaxSuppression",
+    "Col2Im",
+    "CompareAndBitpack",
+    "ComplexAbs",
+    "ConjugateTranspose",
     "Cross",
-    "BondForce",
-    "BondEnergy",
-    "BondAtomEnergy",
-    "BondForceWithAtomEnergy",
-    "BondForceWithAtomVirial",
-    "DihedralForce",
-    "DihedralEnergy",
-    "DihedralAtomEnergy",
-    "DihedralForceWithAtomEnergy",
-    "AngleForce",
-    "AngleEnergy",
-    "AngleAtomEnergy",
-    "AngleForceWithAtomEnergy",
-    'PMEReciprocalForce',
-    'LJForce',
-    'LJForceWithPMEDirectForce',
-    'LJEnergy',
-    'PMEExcludedForce',
-    'PMEEnergy',
-    "Dihedral14LJForce",
-    "Dihedral14LJEnergy",
-    "Dihedral14LJForceWithDirectCF",
-    "Dihedral14LJCFForceWithAtomEnergy",
-    "Dihedral14LJAtomEnergy",
-    "Dihedral14CFEnergy",
-    "MDIterationLeapFrog",
-    "Dihedral14CFAtomEnergy",
-    "MDTemperature",
-    "NeighborListUpdate",
-    "MDIterationLeapFrogLiujian",
-    "CrdToUintCrd",
-    "MDIterationSetupRandState",
-    "TransferCrd",
-    # Update
-    "ConstrainForceCycleWithVirial",
-    "RefreshUintCrd",
-    "LastCrdToDr",
-    "RefreshCrdVel",
-    "CalculateNowrapCrd",
-    "RefreshBoxmapTimes",
-    "Totalc6get",
-    "CrdToUintCrdQuarter",
-    "MDIterationLeapFrogLiujianWithMaxVel",
-    "GetCenterOfMass",
-    "MapCenterOfMass",
-    "NeighborListRefresh",
-    "MDIterationLeapFrog",
-    "MDIterationLeapFrogWithMaxVel",
-    "MDIterationGradientDescent",
-    "BondForceWithAtomEnergyAndVirial",
-    "ConstrainForceCycle",
-    "LJForceWithVirialEnergy",
-    "LJForceWithPMEDirectForceUpdate",
-    "PMEReciprocalForceUpdate",
-    "PMEExcludedForceUpdate",
-    "LJForceWithVirialEnergyUpdate",
-    "Dihedral14ForceWithAtomEnergyVirial",
-    "PMEEnergyUpdate",
-    "ConstrainForceVirial",
-    "ConstrainForce",
-    "Constrain",
+    "Cummin",
+    "CumulativeLogsumexp",
+    "DataFormatVecPermute",
+    "DeformableOffsets",
+    "ExtractImagePatches",
+    "FillDiagonal",
+    "Fills",
+    "Gcd",
+    "Geqrf",
+    "GridSampler2D",
+    "FractionalAvgPool",
+    "FractionalMaxPool",
+    "FractionalMaxPool3DWithFixedKsize",
+    "FractionalMaxPoolWithFixedKsize",
+    "RGBToHSV",
+    "ResizeNearestNeighborV2",
+    "RightShift",
+    "SampleDistortedBoundingBoxV2",
+    "ScaleAndTranslate",
+    "ScatterAddWithAxis",
+    "ScatterNdMul",
+    "SegmentMean",
+    "SegmentProd",
+    "SegmentSum",
+    "SegmentMax",
+    "SegmentMin",
+    "TripletMarginLoss",
+    "UpsampleNearest3D",
+    "UpsampleTrilinear3D",
+    "Trace",
+    "TridiagonalMatMul",
+    "Tril",
+    "TrilIndices",
+    "Triu",
+    "TriuIndices",
+    "UniqueConsecutive",
+    "UnravelIndex",
+    "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__ = [
     "ms_kernel",
-    "ms_hybrid",
+    "kernel",
 ]
 
-__all__.extend(__sponge__)
 __all__.extend(__custom__)
 
 __all__.sort()

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

@@ -13,10 +13,12 @@
 # limitations under the License.
 # ============================================================================
 """cache_ops"""
-from ..._checkparam import Validator as validator
-from ...common import dtype as mstype
-from ..primitive import prim_attr_register, PrimitiveWithCheck
-from .. import signature as sig
+
+from __future__ import absolute_import
+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
 
 
 class UpdateCache(PrimitiveWithCheck):
@@ -103,7 +105,7 @@ class SubAndFilter(PrimitiveWithCheck):
 
 class MapUniform(PrimitiveWithCheck):
     """
-    Map a tensor by using fomula : value = key % `group_num` * `per_group_size` + key // `group_num`.
+    Map a tensor by using formula : value = key % `group_num` * `per_group_size` + key // `group_num`.
 
     Inputs:
         - **input** (Tensor) - Input Tensor.
@@ -256,22 +258,30 @@ class DynamicAssign(PrimitiveWithCheck):
 
 class PadAndShift(PrimitiveWithCheck):
     """
-    Pad a tensor with -1, and shift with a length.
+    Initialize a tensor with -1, and copy a slice from `input_x` to the padded Tensor.
 
-    Inputs:
-        - **input_x** (Tensor) - The input Tensor, which will be copied
-            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.
-        - **shift_idx** (Int) - The idx of cum_sum_arr.
-        if use python, PadAndShift is:
+    Note:
+        If use python, PadAndShift is:
             output = [-1] * cum_sum_arr[-1]
             start = cum_sum_arr[shift_idx]
             end = cum_sum_arr[shift_idx + 1]
             output[start:end] = input_x[:(end-start)]
+
+    Inputs:
+        - **input_x** (Tensor) - The input Tensor, which will be copied
+          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.
+        - **shift_idx** (int) - The idx of `cum_sum_arr` .
+
     Outputs:
-        Tensor, has the same type as original `variable`.
+        - **output** (Tensor) - Tensor, has the same type as `input`.
+
+    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` .
 
     Supported Platforms:
         `CPU`