mindspore 2.6.0rc1__cp310-cp310-win_amd64.whl → 2.7.0rc1__cp310-cp310-win_amd64.whl

mindspore/ops/op_info_register.py

@@ -1354,6 +1354,27 @@ class DataType:
     F64_Default_Tuple = ("float64", "DefaultFormat", "tuple")
     F64_Default_List = ("float64", "DefaultFormat", "list")
 
+    BF16_None = ("bfloat16", "")
+    BF16_Default = ("bfloat16", "DefaultFormat")
+    BF16_5HD = ("bfloat16", "NC1HWC0")
+    BF16_FracZ = ("bfloat16", "FRACTAL_Z")
+    BF16_FracNZ = ("bfloat16", "FRACTAL_NZ")
+    BF16_C1HWNCoC0 = ("bfloat16", "C1HWNCoC0")
+    BF16_NCHW = ("bfloat16", "NCHW")
+    BF16_NHWC = ("bfloat16", "NHWC")
+    BF16_HWCN = ("bfloat16", "HWCN")
+    BF16_NDHWC = ("bfloat16", "NDHWC")
+    BF16_NCDHW = ("bfloat16", "NCDHW")
+    BF16_DHWCN = ("bfloat16", "DHWCN")
+    BF16_NDC1HWC0 = ("bfloat16", "NDC1HWC0")
+    BF16_FRACTAL_Z_3D = ("bfloat16", "FRACTAL_Z_3D")
+    BF16_FracZNLSTM = ("bfloat16", "FRACTAL_ZN_LSTM")
+    BF16_FracZNRNN = ("bfloat16", "FRACTAL_ZN_RNN")
+    BF16_ND_RNNBIAS = ("bfloat16", "ND_RNN_BIAS")
+    BF16_ChannelLast = ("bfloat16", "ChannelLast")
+    BF16_Default_Tuple = ("bfloat16", "DefaultFormat", "tuple")
+    BF16_Default_List = ("bfloat16", "DefaultFormat", "list")
+
     C64_Default = ("complex64", "DefaultFormat")
     C128_Default = ("complex128", "DefaultFormat")
     C64_Default_Tuple = ("complex64", "DefaultFormat", "tuple")

mindspore/ops/operations/__init__.py

@@ -42,25 +42,25 @@ from .array_ops import (ArgMaxWithValue, ArgMinWithValue, Argmax, Argmin, BatchT
                         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, UnsortedSegmentMax, UnsortedSegmentMin, UnsortedSegmentProd,
-                        UnsortedSegmentSum, Unstack, UpperBound, Zeros, ZerosLike, AffineGrid, Bincount, CheckNumerics,
-                        HammingWindow, Identity, 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, IndexPut,
-                        MaskedScatter, TypeAs)
+                        TensorScatterUpdate, TensorShape, Tile, TopK, TransShape, Transpose, TransposeView,
+                        TupleToArray, Unique, UniqueWithPad, UnsortedSegmentMax, UnsortedSegmentMin,
+                        UnsortedSegmentProd, UnsortedSegmentSum, Unstack, UpperBound, Zeros, ZerosLike, AffineGrid,
+                        Bincount, CheckNumerics, HammingWindow, Identity, 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,
+                        IndexPut, MaskedScatter, TypeAs, TransposeExtView)
 from .comm_ops import (AllGather, AllReduce, Reduce, NeighborExchange, NeighborExchangeV2, AlltoAll, _AllSwap,
                        Broadcast, CollectiveGather, CollectiveScatter, Barrier, Send, Receive, BatchISendIRecv,
                        _MirrorOperator, _MirrorMiniStepOperator, _MiniStepAllGather, ReduceOp, _VirtualDataset,
                        _VirtualOutput, _VirtualDiv, _GetTensorSlice, _VirtualAdd, _VirtualAssignAdd, _VirtualAccuGrad,
                        _HostAllGather, _HostReduceScatter, _MirrorMicroStepOperator, _MicroStepAllGather,
-                       _VirtualPipelineEnd, AlltoAllV, ReduceScatter, _VirtualAssignKvCache)
+                       _VirtualPipelineEnd, AlltoAllV, ReduceScatter, _VirtualAssignKvCache, AllGatherV, ReduceScatterV)
 from .control_ops import GeSwitch, Merge
 from .custom_ops import (Custom, CustomOpBuilder)
 from .debug_ops import (ImageSummary, InsertGradientOf, Morph, HookBackward, ScalarSummary,
-                        TensorSummary, HistogramSummary, TensorDump, Print, Assert)
+                        TensorSummary, HistogramSummary, TensorDump, Print, Assert, DumpGradient)
 from .image_ops import (CropAndResize, NonMaxSuppressionV3, HSVToRGB, AdjustHue, AdjustSaturation,
                         NonMaxSuppressionWithOverlaps, ResizeArea, ResizeBilinearV2, ExtractGlimpse,
                         CombinedNonMaxSuppression, RGBToHSV, ScaleAndTranslate, ResizeLinear1D, ResizeBicubic)
@@ -222,6 +222,8 @@ __all__ = [
     'Reshape',
     'Squeeze',
     'Transpose',
+    'TransposeView',
+    'TransposeExtView',
     'OneHot',
     'GatherV2',
     'Gather',
@@ -303,6 +305,7 @@ __all__ = [
     "Print",
     "Assert",
     'InsertGradientOf',
+    'DumpGradient',
     "Morph",
     'HookBackward',
     'InvertPermutation',
@@ -401,6 +404,8 @@ __all__ = [
     'UnsortedSegmentProd',
     "AllGather",
     "AllReduce",
+    "AllGatherV",
+    "ReduceScatterV",
     "Reduce",
     "_AllSwap",
     "ReduceScatter",