forked from mindspore-Ecosystem/mindspore
!45688 expose Primitve APIs for part3 and part4
Merge pull request !45688 from 李林杰/expose_Primitive_APIs_part3_part4
This commit is contained in:
commit
3eea27abbe
|
@ -74,6 +74,8 @@ MindSpore中 `mindspore.ops` 接口与上一版本相比,新增、删除和支
|
||||||
mindspore.ops.Padding
|
mindspore.ops.Padding
|
||||||
mindspore.ops.ResizeNearestNeighbor
|
mindspore.ops.ResizeNearestNeighbor
|
||||||
mindspore.ops.ResizeBilinear
|
mindspore.ops.ResizeBilinear
|
||||||
|
mindspore.ops.UpsampleNearest3D
|
||||||
|
mindspore.ops.UpsampleTrilinear3D
|
||||||
|
|
||||||
损失函数
|
损失函数
|
||||||
^^^^^^^^^^
|
^^^^^^^^^^
|
||||||
|
@ -96,6 +98,7 @@ MindSpore中 `mindspore.ops` 接口与上一版本相比,新增、删除和支
|
||||||
mindspore.ops.SoftMarginLoss
|
mindspore.ops.SoftMarginLoss
|
||||||
mindspore.ops.SoftmaxCrossEntropyWithLogits
|
mindspore.ops.SoftmaxCrossEntropyWithLogits
|
||||||
mindspore.ops.SparseSoftmaxCrossEntropyWithLogits
|
mindspore.ops.SparseSoftmaxCrossEntropyWithLogits
|
||||||
|
mindspore.ops.TripletMarginLoss
|
||||||
|
|
||||||
激活函数
|
激活函数
|
||||||
^^^^^^^^^^
|
^^^^^^^^^^
|
||||||
|
@ -183,6 +186,7 @@ MindSpore中 `mindspore.ops` 接口与上一版本相比,新增、删除和支
|
||||||
mindspore.ops.ComputeAccidentalHits
|
mindspore.ops.ComputeAccidentalHits
|
||||||
mindspore.ops.LogUniformCandidateSampler
|
mindspore.ops.LogUniformCandidateSampler
|
||||||
mindspore.ops.UniformCandidateSampler
|
mindspore.ops.UniformCandidateSampler
|
||||||
|
mindspore.ops.UpsampleNearest3D
|
||||||
|
|
||||||
图像处理
|
图像处理
|
||||||
^^^^^^^^^^
|
^^^^^^^^^^
|
||||||
|
@ -203,7 +207,10 @@ MindSpore中 `mindspore.ops` 接口与上一版本相比,新增、删除和支
|
||||||
mindspore.ops.IOU
|
mindspore.ops.IOU
|
||||||
mindspore.ops.L2Normalize
|
mindspore.ops.L2Normalize
|
||||||
mindspore.ops.NMSWithMask
|
mindspore.ops.NMSWithMask
|
||||||
|
mindspore.ops.RGBToHSV
|
||||||
mindspore.ops.ROIAlign
|
mindspore.ops.ROIAlign
|
||||||
|
mindspore.ops.SampleDistortedBoundingBoxV2
|
||||||
|
mindspore.ops.ScaleAndTranslate
|
||||||
|
|
||||||
文本处理
|
文本处理
|
||||||
^^^^^^^^^^
|
^^^^^^^^^^
|
||||||
|
@ -318,6 +325,7 @@ MindSpore中 `mindspore.ops` 接口与上一版本相比,新增、删除和支
|
||||||
mindspore.ops.TruncateMod
|
mindspore.ops.TruncateMod
|
||||||
mindspore.ops.Xdivy
|
mindspore.ops.Xdivy
|
||||||
mindspore.ops.Xlogy
|
mindspore.ops.Xlogy
|
||||||
|
mindspore.ops.Zeta
|
||||||
|
|
||||||
|
|
||||||
Reduction算子
|
Reduction算子
|
||||||
|
@ -378,6 +386,7 @@ Reduction算子
|
||||||
mindspore.ops.MatMul
|
mindspore.ops.MatMul
|
||||||
mindspore.ops.MatrixInverse
|
mindspore.ops.MatrixInverse
|
||||||
mindspore.ops.Svd
|
mindspore.ops.Svd
|
||||||
|
mindspore.ops.TridiagonalMatMul
|
||||||
|
|
||||||
Tensor操作算子
|
Tensor操作算子
|
||||||
----------------
|
----------------
|
||||||
|
@ -415,6 +424,7 @@ Tensor创建
|
||||||
mindspore.ops.Randperm
|
mindspore.ops.Randperm
|
||||||
mindspore.ops.StandardLaplace
|
mindspore.ops.StandardLaplace
|
||||||
mindspore.ops.StandardNormal
|
mindspore.ops.StandardNormal
|
||||||
|
mindspore.ops.TruncatedNormal
|
||||||
mindspore.ops.UniformInt
|
mindspore.ops.UniformInt
|
||||||
mindspore.ops.UniformReal
|
mindspore.ops.UniformReal
|
||||||
|
|
||||||
|
@ -476,9 +486,22 @@ Array操作
|
||||||
mindspore.ops.Range
|
mindspore.ops.Range
|
||||||
mindspore.ops.Rank
|
mindspore.ops.Rank
|
||||||
mindspore.ops.Reshape
|
mindspore.ops.Reshape
|
||||||
|
mindspore.ops.ResizeNearestNeighborV2
|
||||||
mindspore.ops.ReverseSequence
|
mindspore.ops.ReverseSequence
|
||||||
mindspore.ops.ReverseV2
|
mindspore.ops.ReverseV2
|
||||||
|
mindspore.ops.RightShift
|
||||||
|
mindspore.ops.ScatterAddWithAxis
|
||||||
mindspore.ops.ScatterNd
|
mindspore.ops.ScatterNd
|
||||||
|
mindspore.ops.ScatterNdDiv
|
||||||
|
mindspore.ops.ScatterNdMax
|
||||||
|
mindspore.ops.ScatterNdMin
|
||||||
|
mindspore.ops.ScatterNdMul
|
||||||
|
mindspore.ops.SearchSorted
|
||||||
|
mindspore.ops.SegmentMax
|
||||||
|
mindspore.ops.SegmentMean
|
||||||
|
mindspore.ops.SegmentMin
|
||||||
|
mindspore.ops.SegmentProd
|
||||||
|
mindspore.ops.SegmentSum
|
||||||
mindspore.ops.Select
|
mindspore.ops.Select
|
||||||
mindspore.ops.Shape
|
mindspore.ops.Shape
|
||||||
mindspore.ops.Size
|
mindspore.ops.Size
|
||||||
|
@ -488,8 +511,10 @@ Array操作
|
||||||
mindspore.ops.SpaceToDepth
|
mindspore.ops.SpaceToDepth
|
||||||
mindspore.ops.SparseGatherV2
|
mindspore.ops.SparseGatherV2
|
||||||
mindspore.ops.Split
|
mindspore.ops.Split
|
||||||
|
mindspore.ops.SplitV
|
||||||
mindspore.ops.Squeeze
|
mindspore.ops.Squeeze
|
||||||
mindspore.ops.Stack
|
mindspore.ops.Stack
|
||||||
|
mindspore.ops.STFT
|
||||||
mindspore.ops.StridedSlice
|
mindspore.ops.StridedSlice
|
||||||
mindspore.ops.TensorScatterAdd
|
mindspore.ops.TensorScatterAdd
|
||||||
mindspore.ops.TensorScatterDiv
|
mindspore.ops.TensorScatterDiv
|
||||||
|
@ -500,16 +525,22 @@ Array操作
|
||||||
mindspore.ops.TensorScatterUpdate
|
mindspore.ops.TensorScatterUpdate
|
||||||
mindspore.ops.TensorShape
|
mindspore.ops.TensorShape
|
||||||
mindspore.ops.Tile
|
mindspore.ops.Tile
|
||||||
|
mindspore.ops.Trace
|
||||||
mindspore.ops.Transpose
|
mindspore.ops.Transpose
|
||||||
|
mindspore.ops.Tril
|
||||||
|
mindspore.ops.TrilIndices
|
||||||
|
mindspore.ops.Triu
|
||||||
|
mindspore.ops.TriuIndices
|
||||||
mindspore.ops.Unique
|
mindspore.ops.Unique
|
||||||
|
mindspore.ops.UniqueConsecutive
|
||||||
mindspore.ops.UniqueWithPad
|
mindspore.ops.UniqueWithPad
|
||||||
|
mindspore.ops.UnravelIndex
|
||||||
mindspore.ops.UnsortedSegmentMax
|
mindspore.ops.UnsortedSegmentMax
|
||||||
mindspore.ops.UnsortedSegmentMin
|
mindspore.ops.UnsortedSegmentMin
|
||||||
mindspore.ops.UnsortedSegmentProd
|
mindspore.ops.UnsortedSegmentProd
|
||||||
mindspore.ops.UnsortedSegmentSum
|
mindspore.ops.UnsortedSegmentSum
|
||||||
mindspore.ops.Unstack
|
mindspore.ops.Unstack
|
||||||
mindspore.ops.SplitV
|
mindspore.ops.UpperBound
|
||||||
mindspore.ops.STFT
|
|
||||||
|
|
||||||
类型转换
|
类型转换
|
||||||
^^^^^^^^^^^^^^^^
|
^^^^^^^^^^^^^^^^
|
||||||
|
|
|
@ -74,6 +74,8 @@ Neural Network
|
||||||
mindspore.ops.Padding
|
mindspore.ops.Padding
|
||||||
mindspore.ops.ResizeNearestNeighbor
|
mindspore.ops.ResizeNearestNeighbor
|
||||||
mindspore.ops.ResizeBilinear
|
mindspore.ops.ResizeBilinear
|
||||||
|
mindspore.ops.UpsampleNearest3D
|
||||||
|
mindspore.ops.UpsampleTrilinear3D
|
||||||
|
|
||||||
Loss Function
|
Loss Function
|
||||||
^^^^^^^^^^^^^
|
^^^^^^^^^^^^^
|
||||||
|
@ -96,6 +98,7 @@ Loss Function
|
||||||
mindspore.ops.SoftMarginLoss
|
mindspore.ops.SoftMarginLoss
|
||||||
mindspore.ops.SoftmaxCrossEntropyWithLogits
|
mindspore.ops.SoftmaxCrossEntropyWithLogits
|
||||||
mindspore.ops.SparseSoftmaxCrossEntropyWithLogits
|
mindspore.ops.SparseSoftmaxCrossEntropyWithLogits
|
||||||
|
mindspore.ops.TripletMarginLoss
|
||||||
|
|
||||||
Activation Function
|
Activation Function
|
||||||
^^^^^^^^^^^^^^^^^^^
|
^^^^^^^^^^^^^^^^^^^
|
||||||
|
@ -203,7 +206,10 @@ Image Processing
|
||||||
mindspore.ops.IOU
|
mindspore.ops.IOU
|
||||||
mindspore.ops.L2Normalize
|
mindspore.ops.L2Normalize
|
||||||
mindspore.ops.NMSWithMask
|
mindspore.ops.NMSWithMask
|
||||||
|
mindspore.ops.RGBToHSV
|
||||||
mindspore.ops.ROIAlign
|
mindspore.ops.ROIAlign
|
||||||
|
mindspore.ops.SampleDistortedBoundingBoxV2
|
||||||
|
mindspore.ops.ScaleAndTranslate
|
||||||
|
|
||||||
Text Processing
|
Text Processing
|
||||||
^^^^^^^^^^^^^^^
|
^^^^^^^^^^^^^^^
|
||||||
|
@ -318,6 +324,7 @@ Element-wise Operator
|
||||||
mindspore.ops.TruncateMod
|
mindspore.ops.TruncateMod
|
||||||
mindspore.ops.Xdivy
|
mindspore.ops.Xdivy
|
||||||
mindspore.ops.Xlogy
|
mindspore.ops.Xlogy
|
||||||
|
mindspore.ops.Zeta
|
||||||
|
|
||||||
|
|
||||||
Reduction Operator
|
Reduction Operator
|
||||||
|
@ -378,6 +385,7 @@ Linear Algebraic Operator
|
||||||
mindspore.ops.MatMul
|
mindspore.ops.MatMul
|
||||||
mindspore.ops.MatrixInverse
|
mindspore.ops.MatrixInverse
|
||||||
mindspore.ops.Svd
|
mindspore.ops.Svd
|
||||||
|
mindspore.ops.TridiagonalMatMul
|
||||||
|
|
||||||
Tensor Operation Operator
|
Tensor Operation Operator
|
||||||
--------------------------
|
--------------------------
|
||||||
|
@ -415,6 +423,7 @@ Random Generation Operator
|
||||||
mindspore.ops.Randperm
|
mindspore.ops.Randperm
|
||||||
mindspore.ops.StandardLaplace
|
mindspore.ops.StandardLaplace
|
||||||
mindspore.ops.StandardNormal
|
mindspore.ops.StandardNormal
|
||||||
|
mindspore.ops.TruncatedNormal
|
||||||
mindspore.ops.UniformInt
|
mindspore.ops.UniformInt
|
||||||
mindspore.ops.UniformReal
|
mindspore.ops.UniformReal
|
||||||
|
|
||||||
|
@ -476,9 +485,22 @@ Array Operation
|
||||||
mindspore.ops.Range
|
mindspore.ops.Range
|
||||||
mindspore.ops.Rank
|
mindspore.ops.Rank
|
||||||
mindspore.ops.Reshape
|
mindspore.ops.Reshape
|
||||||
|
mindspore.ops.ResizeNearestNeighborV2
|
||||||
mindspore.ops.ReverseSequence
|
mindspore.ops.ReverseSequence
|
||||||
mindspore.ops.ReverseV2
|
mindspore.ops.ReverseV2
|
||||||
|
mindspore.ops.RightShift
|
||||||
|
mindspore.ops.ScatterAddWithAxis
|
||||||
mindspore.ops.ScatterNd
|
mindspore.ops.ScatterNd
|
||||||
|
mindspore.ops.ScatterNdDiv
|
||||||
|
mindspore.ops.ScatterNdMax
|
||||||
|
mindspore.ops.ScatterNdMin
|
||||||
|
mindspore.ops.ScatterNdMul
|
||||||
|
mindspore.ops.SearchSorted
|
||||||
|
mindspore.ops.SegmentMax
|
||||||
|
mindspore.ops.SegmentMean
|
||||||
|
mindspore.ops.SegmentMin
|
||||||
|
mindspore.ops.SegmentProd
|
||||||
|
mindspore.ops.SegmentSum
|
||||||
mindspore.ops.Select
|
mindspore.ops.Select
|
||||||
mindspore.ops.Shape
|
mindspore.ops.Shape
|
||||||
mindspore.ops.Size
|
mindspore.ops.Size
|
||||||
|
@ -488,8 +510,10 @@ Array Operation
|
||||||
mindspore.ops.SpaceToDepth
|
mindspore.ops.SpaceToDepth
|
||||||
mindspore.ops.SparseGatherV2
|
mindspore.ops.SparseGatherV2
|
||||||
mindspore.ops.Split
|
mindspore.ops.Split
|
||||||
|
mindspore.ops.SplitV
|
||||||
mindspore.ops.Squeeze
|
mindspore.ops.Squeeze
|
||||||
mindspore.ops.Stack
|
mindspore.ops.Stack
|
||||||
|
mindspore.ops.STFT
|
||||||
mindspore.ops.StridedSlice
|
mindspore.ops.StridedSlice
|
||||||
mindspore.ops.TensorScatterAdd
|
mindspore.ops.TensorScatterAdd
|
||||||
mindspore.ops.TensorScatterDiv
|
mindspore.ops.TensorScatterDiv
|
||||||
|
@ -500,16 +524,22 @@ Array Operation
|
||||||
mindspore.ops.TensorScatterUpdate
|
mindspore.ops.TensorScatterUpdate
|
||||||
mindspore.ops.TensorShape
|
mindspore.ops.TensorShape
|
||||||
mindspore.ops.Tile
|
mindspore.ops.Tile
|
||||||
|
mindspore.ops.Trace
|
||||||
mindspore.ops.Transpose
|
mindspore.ops.Transpose
|
||||||
|
mindspore.ops.Tril
|
||||||
|
mindspore.ops.TrilIndices
|
||||||
|
mindspore.ops.Triu
|
||||||
|
mindspore.ops.TriuIndices
|
||||||
mindspore.ops.Unique
|
mindspore.ops.Unique
|
||||||
|
mindspore.ops.UniqueConsecutive
|
||||||
mindspore.ops.UniqueWithPad
|
mindspore.ops.UniqueWithPad
|
||||||
|
mindspore.ops.UnravelIndex
|
||||||
mindspore.ops.UnsortedSegmentMax
|
mindspore.ops.UnsortedSegmentMax
|
||||||
mindspore.ops.UnsortedSegmentMin
|
mindspore.ops.UnsortedSegmentMin
|
||||||
mindspore.ops.UnsortedSegmentProd
|
mindspore.ops.UnsortedSegmentProd
|
||||||
mindspore.ops.UnsortedSegmentSum
|
mindspore.ops.UnsortedSegmentSum
|
||||||
mindspore.ops.Unstack
|
mindspore.ops.Unstack
|
||||||
mindspore.ops.SplitV
|
mindspore.ops.UpperBound
|
||||||
mindspore.ops.STFT
|
|
||||||
|
|
||||||
Type Conversion
|
Type Conversion
|
||||||
^^^^^^^^^^^^^^^
|
^^^^^^^^^^^^^^^
|
||||||
|
|
|
@ -46,7 +46,9 @@ from .array_ops import (ArgMaxWithValue, ArgMinWithValue, Argmax, Argmin, BatchT
|
||||||
UnsortedSegmentSum, Unstack, UpperBound, Zeros, ZerosLike, AffineGrid, Bincount, CheckNumerics,
|
UnsortedSegmentSum, Unstack, UpperBound, Zeros, ZerosLike, AffineGrid, Bincount, CheckNumerics,
|
||||||
HammingWindow, IdentityN, IndexFill, LeftShift, ListDiff, LogSpace, MatrixBandPart,
|
HammingWindow, IdentityN, IndexFill, LeftShift, ListDiff, LogSpace, MatrixBandPart,
|
||||||
MatrixDiagPartV3, MatrixDiagV3, MatrixSetDiagV3, NonZero, Expand, Col2Im, ConjugateTranspose,
|
MatrixDiagPartV3, MatrixDiagV3, MatrixSetDiagV3, NonZero, Expand, Col2Im, ConjugateTranspose,
|
||||||
FillDiagonal, Fills)
|
FillDiagonal, Fills, ResizeNearestNeighborV2, RightShift, ScatterAddWithAxis,
|
||||||
|
ScatterNdMul, SegmentMean, SegmentProd, SegmentSum, SegmentMax, SegmentMin, Tril, Triu,
|
||||||
|
UniqueConsecutive, UnravelIndex)
|
||||||
from .comm_ops import (AllGather, AllReduce, NeighborExchange, NeighborExchangeV2, AlltoAll, _AllSwap, ReduceScatter,
|
from .comm_ops import (AllGather, AllReduce, NeighborExchange, NeighborExchangeV2, AlltoAll, _AllSwap, ReduceScatter,
|
||||||
Broadcast,
|
Broadcast,
|
||||||
_MirrorOperator, _MirrorMiniStepOperator, _MiniStepAllGather, ReduceOp, _VirtualDataset,
|
_MirrorOperator, _MirrorMiniStepOperator, _MiniStepAllGather, ReduceOp, _VirtualDataset,
|
||||||
|
@ -59,7 +61,7 @@ from .debug_ops import (ImageSummary, InsertGradientOf, HookBackward, ScalarSumm
|
||||||
TensorSummary, HistogramSummary, Print, Assert)
|
TensorSummary, HistogramSummary, Print, Assert)
|
||||||
from .image_ops import (CropAndResize, NonMaxSuppressionV3, HSVToRGB, AdjustHue, AdjustSaturation,
|
from .image_ops import (CropAndResize, NonMaxSuppressionV3, HSVToRGB, AdjustHue, AdjustSaturation,
|
||||||
NonMaxSuppressionWithOverlaps, ResizeArea, ResizeBicubic, ResizeBilinearV2, ExtractGlimpse,
|
NonMaxSuppressionWithOverlaps, ResizeArea, ResizeBicubic, ResizeBilinearV2, ExtractGlimpse,
|
||||||
CombinedNonMaxSuppression)
|
CombinedNonMaxSuppression, RGBToHSV, ScaleAndTranslate)
|
||||||
from .inner_ops import (ScalarCast, Randperm, NoRepeatNGram, LambApplyOptimizerAssign, LambApplyWeightAssign,
|
from .inner_ops import (ScalarCast, Randperm, NoRepeatNGram, LambApplyOptimizerAssign, LambApplyWeightAssign,
|
||||||
FusedWeightScaleApplyMomentum, FusedCastAdamWeightDecay, FusedAdaFactor,
|
FusedWeightScaleApplyMomentum, FusedCastAdamWeightDecay, FusedAdaFactor,
|
||||||
FusedAdaFactorWithGlobalNorm)
|
FusedAdaFactorWithGlobalNorm)
|
||||||
|
@ -83,7 +85,7 @@ from .math_ops import (Abs, ACos, Asin, Asinh, AddN, AccumulateNV2, AssignAdd, A
|
||||||
FFTWithSize, Heaviside, Histogram, Hypot, Lcm, LuUnpack, MatrixExp,
|
FFTWithSize, Heaviside, Histogram, Hypot, Lcm, LuUnpack, MatrixExp,
|
||||||
MatrixLogarithm, MatrixPower, MatrixSolve, MatrixTriangularSolve, ReduceStd, STFT,
|
MatrixLogarithm, MatrixPower, MatrixSolve, MatrixTriangularSolve, ReduceStd, STFT,
|
||||||
NextAfter, Orgqr, Qr, RaggedRange, Digamma, Eig, EuclideanNorm, CompareAndBitpack, ComplexAbs,
|
NextAfter, Orgqr, Qr, RaggedRange, Digamma, Eig, EuclideanNorm, CompareAndBitpack, ComplexAbs,
|
||||||
CumulativeLogsumexp, Gcd)
|
CumulativeLogsumexp, Gcd, Trace, TridiagonalMatMul, TrilIndices, TriuIndices, Zeta)
|
||||||
from .nn_ops import (LSTM, SGD, Adam, AdamWeightDecay, FusedSparseAdam, FusedSparseLazyAdam, AdamNoUpdateParam,
|
from .nn_ops import (LSTM, SGD, Adam, AdamWeightDecay, FusedSparseAdam, FusedSparseLazyAdam, AdamNoUpdateParam,
|
||||||
ApplyMomentum, BatchNorm, BiasAdd, Conv2D, Conv3D, Conv2DTranspose, Conv3DTranspose,
|
ApplyMomentum, BatchNorm, BiasAdd, Conv2D, Conv3D, Conv2DTranspose, Conv3DTranspose,
|
||||||
DepthwiseConv2dNative,
|
DepthwiseConv2dNative,
|
||||||
|
@ -110,10 +112,11 @@ from .nn_ops import (LSTM, SGD, Adam, AdamWeightDecay, FusedSparseAdam, FusedSpa
|
||||||
GridSampler3D, MaxPool3DWithArgmax, MaxUnpool2D, NuclearNorm, NthElement, MultilabelMarginLoss,
|
GridSampler3D, MaxPool3DWithArgmax, MaxUnpool2D, NuclearNorm, NthElement, MultilabelMarginLoss,
|
||||||
PSROIPooling, Dilation2D, DataFormatVecPermute, DeformableOffsets, FractionalAvgPool,
|
PSROIPooling, Dilation2D, DataFormatVecPermute, DeformableOffsets, FractionalAvgPool,
|
||||||
FractionalMaxPool, FractionalMaxPool3DWithFixedKsize, FractionalMaxPoolWithFixedKsize,
|
FractionalMaxPool, FractionalMaxPool3DWithFixedKsize, FractionalMaxPoolWithFixedKsize,
|
||||||
GridSampler2D)
|
GridSampler2D, TripletMarginLoss, UpsampleNearest3D, UpsampleTrilinear3D)
|
||||||
from .other_ops import (Assign, IOU, BartlettWindow, BlackmanWindow, BoundingBoxDecode, BoundingBoxEncode,
|
from .other_ops import (Assign, IOU, BartlettWindow, BlackmanWindow, BoundingBoxDecode, BoundingBoxEncode,
|
||||||
ConfusionMatrix, UpdateState, Load, StopGradient,
|
ConfusionMatrix, UpdateState, Load, StopGradient,
|
||||||
CheckValid, Partial, Depend, identity, Push, Pull, PyFunc, _DynamicLossScale)
|
CheckValid, Partial, Depend, identity, Push, Pull, PyFunc, _DynamicLossScale,
|
||||||
|
SampleDistortedBoundingBoxV2)
|
||||||
from .random_ops import (RandomChoiceWithMask, StandardNormal, Gamma, RandomGamma, Poisson, UniformInt, UniformReal,
|
from .random_ops import (RandomChoiceWithMask, StandardNormal, Gamma, RandomGamma, Poisson, UniformInt, UniformReal,
|
||||||
RandomCategorical, StandardLaplace, Multinomial, UniformCandidateSampler,
|
RandomCategorical, StandardLaplace, Multinomial, UniformCandidateSampler,
|
||||||
LogUniformCandidateSampler, TruncatedNormal, LogNormalReverse, NonDeterministicInts,
|
LogUniformCandidateSampler, TruncatedNormal, LogNormalReverse, NonDeterministicInts,
|
||||||
|
@ -586,7 +589,6 @@ __all__ = [
|
||||||
"ParameterizedTruncatedNormal",
|
"ParameterizedTruncatedNormal",
|
||||||
"RandomGamma",
|
"RandomGamma",
|
||||||
"RandomPoisson",
|
"RandomPoisson",
|
||||||
"PadAndShift",
|
|
||||||
"RaggedRange",
|
"RaggedRange",
|
||||||
"Digamma",
|
"Digamma",
|
||||||
"Dilation2D",
|
"Dilation2D",
|
||||||
|
@ -614,6 +616,30 @@ __all__ = [
|
||||||
"FractionalMaxPool",
|
"FractionalMaxPool",
|
||||||
"FractionalMaxPool3DWithFixedKsize",
|
"FractionalMaxPool3DWithFixedKsize",
|
||||||
"FractionalMaxPoolWithFixedKsize",
|
"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",
|
||||||
]
|
]
|
||||||
|
|
||||||
__custom__ = [
|
__custom__ = [
|
||||||
|
|
Loading…
Reference in New Issue