Operator List

View Source On Gitee

mindspore.nn

Operation

Ascend

GPU

CPU

Operator Type

mindspore.nn.Softmax

Supported

Supported

Supported

layer/activation

mindspore.nn.LogSoftmax

Supported

Supported

Doing

layer/activation

mindspore.nn.ReLU

Supported

Supported

Supported

layer/activation

mindspore.nn.ReLU6

Supported

Supported

Doing

layer/activation

mindspore.nn.HSwish

Doing

Supported

Doing

layer/activation

mindspore.nn.HSigmoid

Doing

Supported

Doing

layer/activation

mindspore.nn.LeakyReLU

Supported

Doing

Doing

layer/activation

mindspore.nn.Tanh

Supported

Supported

Doing

layer/activation

mindspore.nn.GELU

Supported

Supported

Doing

layer/activation

mindspore.nn.Sigmoid

Supported

Supported

Doing

layer/activation

mindspore.nn.PReLU

Supported

Doing

Doing

layer/activation

mindspore.nn.Dropout

Supported

Supported

Doing

layer/basic

mindspore.nn.Flatten

Supported

Supported

Doing

layer/basic

mindspore.nn.Dense

Supported

Supported

Doing

layer/basic

mindspore.nn.DenseBnAct

Supported

Doing

Supported

layer/basic

mindspore.nn.ClipByNorm

Doing

Supported

Doing

layer/basic

mindspore.nn.Norm

Doing

Supported

Doing

layer/basic

mindspore.nn.OneHot

Supported

Supported

Doing

layer/basic

mindspore.nn.Range

Supported

Doing

Doing

layer/basic

mindspore.nn.SequentialCell

Supported

Supported

Doing

layer/container

mindspore.nn.CellList

Supported

Supported

Doing

layer/container

mindspore.nn.Conv2d

Supported

Supported

Supported

layer/conv

mindspore.nn.Conv2dTranspose

Supported

Supported

Doing

layer/conv

mindspore.nn.Conv2dBnAct

Supported

Supported

Supported

layer/conv

mindspore.nn.Conv1d

Supported

Doing

Doing

layer/conv

mindspore.nn.Conv1dTranspose

Supported

Doing

Doing

layer/conv

mindspore.nn.Embedding

Doing

Supported

Doing

layer/embedding

mindspore.nn.ImageGradients

Doing

Doing

Doing

layer/image

mindspore.nn.SSIM

Doing

Doing

Doing

layer/image

mindspore.nn.PSNR

Doing

Doing

Doing

layer/image

mindspore.nn.CentralCrop

Supported

Doing

Doing

layer/image

mindspore.nn.LSTM

Doing

Supported

Supported

layer/lstm

mindspore.nn.GlobalBatchNorm

Supported

Doing

Doing

layer/normalization

mindspore.nn.BatchNorm1d

Supported

Doing

Doing

layer/normalization

mindspore.nn.BatchNorm2d

Supported

Supported

Doing

layer/normalization

mindspore.nn.GroupNorm

Supported

Doing

Doing

layer/normalization

mindspore.nn.LayerNorm

Supported

Supported

Doing

layer/normalization

mindspore.nn.MatrixDiag

Supported

Doing

Doing

layer/normalization

mindspore.nn.MatrixDiagPart

Supported

Doing

Doing

layer/normalization

mindspore.nn.MatrixSetDiag

Supported

Doing

Doing

layer/normalization

mindspore.nn.LinSpace

Supported

Doing

Doing

layer/normalization

mindspore.nn.MaxPool2d

Supported

Supported

Supported

layer/pooling

mindspore.nn.AvgPool2d

Doing

Supported

Doing

layer/pooling

mindspore.nn.L1Loss

Doing

Doing

Doing

loss/loss

mindspore.nn.MSELoss

Supported

Doing

Doing

loss/loss

mindspore.nn.SmoothL1Loss

Supported

Doing

Doing

loss/loss

mindspore.nn.SoftmaxCrossEntropyWithLogits

Supported

Supported

Doing

loss/loss

mindspore.nn.SoftmaxCrossEntropyExpand

Supported

Doing

Doing

loss/loss

mindspore.nn.ProximalAdagrad

Supported

Doing

Doing

optim/ProximalAdagrad

mindspore.nn.LazyAdam

Supported

Doing

Doing

optim/lazyadam

mindspore.nn.Adam

Supported

Doing

Doing

optim/adam

mindspore.nn.AdamWeightDecay

Supported

Supported

Doing

optim/adam

mindspore.nn.Lamb

Supported

Supported

Doing

optim/lamb

mindspore.nn.LARS

Supported

Doing

Doing

optim/lars

mindspore.nn.Momentum

Supported

Supported

Doing

optim/momentum

mindspore.nn.Optimizer

Supported

Supported

Doing

optim/optimizer

mindspore.nn.RMSProp

Supported

Support

Doing

optim/optimizer

mindspore.nn.SGD

Supported

Doing

Doing

optim/sgd

mindspore.nn.WithLossCell

Supported

Supported

Doing

wrap/cell_wrapper

mindspore.nn.WithGradCell

Supported

Supported

Doing

wrap/cell_wrapper

mindspore.nn.TrainOneStepCell

Supported

Supported

Doing

wrap/cell_wrapper

mindspore.nn.DataWrapper

Doing

Supported

Doing

wrap/cell_wrapper

mindspore.nn.GetNextSingleOp

Doing

Supported

Doing

wrap/cell_wrapper

mindspore.nn.WithEvalCell

Supported

Supported

Doing

wrap/cell_wrapper

mindspore.nn.ParameterUpdate

Supported

Doing

Doing

wrap/cell_wrapper

mindspore.nn.DistributedGradReducer

Supported

Doing

Doing

wrap/grad_reducer

mindspore.nn.DynamicLossScaleUpdateCell

Doing

Doing

Doing

wrap/loss_scale

mindspore.nn.FixedLossScaleUpdateCell

Doing

Doing

Doing

wrap/loss_scale

mindspore.nn.TrainOneStepWithLossScaleCell

Doing

Doing

Doing

wrap/loss_scale

mindspore.nn.Cell

Supported

Supported

Supported

cell

mindspore.ops.operations

Operation

Ascend

GPU

CPU

Operator Type

mindspore.ops.operations.Flatten

Supported

Supported

Doing

nn_ops

mindspore.ops.operations.Softmax

Supported

Supported

Supported

nn_ops

mindspore.ops.operations.Acosh

Doing

Doing

Doing

nn_ops

mindspore.ops.operations.FloorMod

Supported

Doing

Doing

nn_ops

mindspore.ops.operations.Elu

Supported

Doing

Doing

nn_ops

mindspore.ops.operations.MirrorPad

Doing

Doing

Doing

nn_ops

mindspore.ops.operations.Unpack

Supported

Doing

Doing

nn_ops

mindspore.ops.operations.Pack

Supported

Doing

Doing

nn_ops

mindspore.ops.operations.L2Loss

Supported

Doing

Doing

nn_ops

mindspore.ops.operations.CTCLoss

Supported

Doing

Doing

nn_ops

mindspore.ops.operations.RNNTLoss

Supported

Doing

Doing

nn_ops

mindspore.ops.operations.LogSoftmax

Supported

Supported

Doing

nn_ops

mindspore.ops.operations.Softplus

Supported

Doing

Doing

nn_ops

mindspore.ops.operations.ReLU

Supported

Supported

Supported

nn_ops

mindspore.ops.operations.ReLU6

Supported

Supported

Doing

nn_ops

mindspore.ops.operations.HSwish

Doing

Supported

Doing

nn_ops

mindspore.ops.operations.HSigmoid

Doing

Supported

Doing

nn_ops

mindspore.ops.operations.Sigmoid

Supported

Supported

Doing

nn_ops

mindspore.ops.operations.Tanh

Supported

Supported

Doing

nn_ops

mindspore.ops.operations.BatchNorm

Supported

Doing

Doing

nn_ops

mindspore.ops.operations.LRN

Supported

Doing

Doing

nn_ops

mindspore.ops.operations.Conv2D

Supported

Supported

Supported

nn_ops

mindspore.ops.operations.DepthwiseConv2dNative

Supported

Doing

Doing

nn_ops

mindspore.ops.operations.DepthwiseConv2dNativeBackpropInput

Supported

Doing

Doing

nn_ops

mindspore.ops.operations.DepthwiseConv2dNativeiBackpropFilter

Supported

Doing

Doing

nn_ops

mindspore.ops.operations.MaxPoolWithArgmax

Supported

Doing

Doing

nn_ops

mindspore.ops.operations.MaxPool

Supported

Supported

Supported

nn_ops

mindspore.ops.operations.AvgPool

Supported

Supported

Doing

nn_ops

mindspore.ops.operations.Conv2DBackpropInput

Supported

Supported

Doing

nn_ops

mindspore.ops.operations.BiasAdd

Supported

Supported

Supported

nn_ops

mindspore.ops.operations.TopK

Supported

Doing

Doing

nn_ops

mindspore.ops.operations.SoftmaxCrossEntropyWithLogits

Supported

Supported

Doing

nn_ops

mindspore.ops.operations.SparseSoftmaxCrossEntropyWithLogits

Doing

Supported

Supported

nn_ops

mindspore.ops.operations.ApplyMomentum

Supported

Supported

Supported

nn_ops

mindspore.ops.operations.ApplyAddSign

Supported

Doing

Doing

nn_ops

mindspore.ops.operations.ApplyPowerSign

Supported

Doing

Doing

nn_ops

mindspore.ops.operations.ApplyGradientDescent

Supported

Doing

Doing

nn_ops

mindspore.ops.operations.ApplyProximalGradientDescent

Supported

Doing

Doing

nn_ops

mindspore.ops.operations.ApplyRMSProp

Supported

Supported

Doing

nn_ops

mindspore.ops.operations.ApplyCenteredRMSProp

Supported

Supported

Doing

nn_ops

mindspore.ops.operations.SparseApplyAdagrad

Supported

Doing

Doing

nn_ops

mindspore.ops.operations.SparseApplyAdagradV2

Supported

Doing

Doing

nn_ops

mindspore.ops.operations.SparseApplyProximalAdagrad

Supported

Doing

Doing

nn_ops

mindspore.ops.operations.FusedSparseProximalAdagrad

Doing

Doing

Supported

nn_ops

mindspore.ops.operations.ApplyProximalAdagrad

Supported

Doing

Doing

nn_ops

mindspore.ops.operations.FusedSparseLazyAdam

Doing

Doing

Supported

nn_ops

mindspore.ops.operations.FusedSparseAdam

Doing

Doing

Supported

nn_ops

mindspore.ops.operations.SmoothL1Loss

Supported

Doing

Doing

nn_ops

mindspore.ops.operations.SGD

Supported

Doing

Doing

nn_ops

mindspore.ops.operations.LayerNorm

Supported

Supported

Doing

nn_ops

mindspore.ops.operations.L2Normalize

Supported

Doing

Doing

nn_ops

mindspore.ops.operations.DropoutGenMask

Supported

Doing

Doing

nn_ops

mindspore.ops.operations.DropoutDoMask

Supported

Doing

Doing

nn_ops

mindspore.ops.operations.ResizeBilinear

Supported

Doing

Doing

nn_ops

mindspore.ops.operations.OneHot

Supported

Supported

Doing

nn_ops

mindspore.ops.operations.Gelu

Supported

Supported

Doing

nn_ops

mindspore.ops.operations.GetNext

Supported

Supported

Doing

nn_ops

mindspore.ops.operations.PReLU

Supported

Doing

Doing

nn_ops

mindspore.ops.operations.LSTM

Doing

Supported

Supported

nn_ops

mindspore.ops.operations.BasicLSTMCell

Doing

Doing

Doing

nn_ops

mindspore.ops.operations.SigmoidCrossEntropyWithLogits

Supported

Supported

Doing

nn_ops

mindspore.ops.operations.Pad

Supported

Doing

Doing

nn_ops

mindspore.ops.operations.ROIAlign

Supported

Doing

Doing

nn_ops

mindspore.ops.operations.Adam

Supported

Supported

Doing

nn_ops

mindspore.ops.operations.BinaryCrossEntropy

Supported

Doing

Doing

nn_ops

mindspore.ops.operations.LARSUpdate

Supported

Doing

Doing

nn_ops

mindspore.ops.operations.Softsign

Supported

Doing

Doing

nn_ops

mindspore.ops.operations.TensorAdd

Supported

Supported

Doing

math_ops

mindspore.ops.operations.AssignAdd

Supported

Supported

Doing

math_ops

mindspore.ops.operations.AssignSub

Supported

Doing

Doing

math_ops

mindspore.ops.operations.ReduceMean

Supported

Supported

Doing

math_ops

mindspore.ops.operations.ReduceSum

Supported

Supported

Doing

math_ops

mindspore.ops.operations.ReduceAll

Supported

Doing

Doing

math_ops

mindspore.ops.operations.ReduceMax

Supported

Supported

Doing

math_ops

mindspore.ops.operations.ReduceMin

Supported

Doing

Doing

math_ops

mindspore.ops.operations.ReduceProd

Supported

Doing

Doing

math_ops

mindspore.ops.operations.CumProd

Supported

Doing

Doing

math_ops

mindspore.ops.operations.MatMul

Supported

Supported

Supported

math_ops

mindspore.ops.operations.BatchMatMul

Supported

Supported

Doing

math_ops

mindspore.ops.operations.CumSum

Supported

Doing

Doing

math_ops

mindspore.ops.operations.AddN

Supported

Supported

Doing

math_ops

mindspore.ops.operations.Neg

Supported

Supported

Doing

math_ops

mindspore.ops.operations.Sub

Supported

Supported

Doing

math_ops

mindspore.ops.operations.Mul

Supported

Supported

Supported

math_ops

mindspore.ops.operations.Square

Supported

Supported

Doing

math_ops

mindspore.ops.operations.SquareSumAll

Supported

Doing

Doing

math_ops

mindspore.ops.operations.Rsqrt

Supported

Supported

Doing

math_ops

mindspore.ops.operations.Sqrt

Supported

Supported

Doing

math_ops

mindspore.ops.operations.Reciprocal

Supported

Supported

Doing

math_ops

mindspore.ops.operations.Pow

Supported

Supported

Doing

math_ops

mindspore.ops.operations.Exp

Supported

Supported

Doing

math_ops

mindspore.ops.operations.Log

Supported

Supported

Doing

math_ops

mindspore.ops.operations.Log1p

Supported

Doing

Doing

math_ops

mindspore.ops.operations.Minimum

Supported

Supported

Doing

math_ops

mindspore.ops.operations.Maximum

Supported

Supported

Doing

math_ops

mindspore.ops.operations.RealDiv

Supported

Supported

Doing

math_ops

mindspore.ops.operations.Div

Supported

Doing

Doing

math_ops

mindspore.ops.operations.DivNoNan

Supported

Doing

Doing

math_ops

mindspore.ops.operations.FloorDiv

Supported

Doing

Doing

math_ops

mindspore.ops.operations.Floor

Supported

Doing

Doing

math_ops

mindspore.ops.operations.Equal

Supported

Supported

Doing

math_ops

mindspore.ops.operations.EqualCount

Doing

Supported

Supported

math_ops

mindspore.ops.operations.NotEqual

Supported

Supported

Doing

math_ops

mindspore.ops.operations.Greater

Supported

Supported

Doing

math_ops

mindspore.ops.operations.GreaterEqual

Supported

Supported

Doing

math_ops

mindspore.ops.operations.Less

Supported

Supported

Doing

math_ops

mindspore.ops.operations.Atan2

Supported

Doing

Doing

math_ops

mindspore.ops.operations.LessEqual

Supported

Supported

Doing

math_ops

mindspore.ops.operations.LogicalNot

Supported

Supported

Doing

math_ops

mindspore.ops.operations.LogicalAnd

Supported

Supported

Doing

math_ops

mindspore.ops.operations.LogicalOr

Supported

Supported

Doing

math_ops

mindspore.ops.operations.BitwiseAnd

Supported

Doing

Doing

math_ops

mindspore.ops.operations.BitwiseOr

Supported

Doing

Doing

math_ops

mindspore.ops.operations.Ceil

Supported

Doing

Doing

math_ops

mindspore.ops.operations.Inv

Supported

Doing

Doing

math_ops

mindspore.ops.operations.Invert

Supported

Doing

Doing

math_ops

mindspore.ops.operations.BitwiseXor

Supported

Doing

Doing

math_ops

mindspore.ops.operations.NPUAllocFloatStatus

Supported

Doing

Doing

math_ops

mindspore.ops.operations.NPUGetFloatStatus

Supported

Doing

Doing

math_ops

mindspore.ops.operations.NPUClearFloatStatus

Supported

Doing

Doing

math_ops

mindspore.ops.operations.FloatStatus

Doing

Supported

Doing

math_ops

mindspore.ops.operations.Cos

Supported

Doing

Doing

math_ops

mindspore.ops.operations.Cosh

Supported

Doing

Doing

math_ops

mindspore.ops.operations.ACos

Supported

Doing

Doing

math_ops

mindspore.ops.operations.BesselI0e

Supported

Doing

Doing

math_ops

mindspore.ops.operations.BesselI1e

Supported

Doing

Doing

math_ops

mindspore.ops.operations.TruncateDiv

Supported

Doing

Doing

math_ops

mindspore.ops.operations.TruncateMod

Supported

Doing

Doing

math_ops

mindspore.ops.operations.Tan

Supported

Doing

Doing

math_ops

mindspore.ops.operations.Asin

Supported

Doing

Doing

math_ops

mindspore.ops.operations.Asinh

Supported

Doing

Doing

math_ops

mindspore.ops.operations.Erf

Supported

Doing

Doing

math_ops

mindspore.ops.operations.Erfc

Supported

Doing

Doing

math_ops

mindspore.ops.operations.Sin

Supported

Doing

Doing

math_ops

mindspore.ops.operations.Sinh

Supported

Doing

Doing

math_ops

mindspore.ops.operations.Expm1

Supported

Doing

Doing

math_ops

mindspore.ops.operations.NMSWithMask

Supported

Doing

Doing

math_ops

mindspore.ops.operations.Abs

Supported

Doing

Doing

math_ops

mindspore.ops.operations.Sign

Supported

Doing

Doing

math_ops

mindspore.ops.operations.Round

Supported

Doing

Doing

math_ops

mindspore.ops.operations.ApproximateEqual

Supported

Doing

Doing

math_ops

mindspore.ops.operations.InplaceAdd

Supported

Doing

Doing

math_ops

mindspore.ops.operations.InplaceSub

Supported

Doing

Doing

math_ops

mindspore.ops.operations.Mod

Supported

Doing

Doing

math_ops

mindspore.ops.operations.ExpandDims

Supported

Supported

Doing

array_ops

mindspore.ops.operations.DType

Supported

Supported

Supported

array_ops

mindspore.ops.operations.SameTypeShape

Supported

Supported

Supported

array_ops

mindspore.ops.operations.Cast

Supported

Supported

Doing

array_ops

mindspore.ops.operations.IsSubClass

Supported

Supported

Supported

array_ops

mindspore.ops.operations.IsInstance

Supported

Supported

Supported

array_ops

mindspore.ops.operations.Reshape

Supported

Supported

Supported

array_ops

mindspore.ops.operations.Shape

Supported

Supported

Supported

array_ops

mindspore.ops.operations.Squeeze

Supported

Supported

Doing

array_ops

mindspore.ops.operations.Transpose

Supported

Supported

Doing

array_ops

mindspore.ops.operations.GatherV2

Supported

Supported

Doing

array_ops

mindspore.ops.operations.Split

Supported

Supported

Doing

array_ops

mindspore.ops.operations.Rank

Supported

Supported

Supported

array_ops

mindspore.ops.operations.TruncatedNormal

Doing

Doing

Doing

array_ops

mindspore.ops.operations.Size

Supported

Supported

Supported

array_ops

mindspore.ops.operations.Fill

Supported

Supported

Supported

array_ops

mindspore.ops.operations.OnesLike

Supported

Doing

Doing

array_ops

mindspore.ops.operations.ZerosLike

Supported

Supported

Doing

array_ops

mindspore.ops.operations.TupleToArray

Supported

Supported

Supported

array_ops

mindspore.ops.operations.ScalarToArray

Supported

Supported

Supported

array_ops

mindspore.ops.operations.ScalarToTensor

Supported

Supported

Supported

array_ops

mindspore.ops.operations.InvertPermutation

Supported

Doing

Doing

array_ops

mindspore.ops.operations.Argmax

Supported

Supported

Supported

array_ops

mindspore.ops.operations.Argmin

Supported

Doing

Doing

array_ops

mindspore.ops.operations.ArgMaxWithValue

Supported

Supported

Doing

array_ops

mindspore.ops.operations.ArgMinWithValue

Supported

Doing

Doing

array_ops

mindspore.ops.operations.Tile

Supported

Supported

Doing

array_ops

mindspore.ops.operations.UnsortedSegmentSum

Supported

Supported

Doing

array_ops

mindspore.ops.operations.UnsortedSegmentMin

Supported

Doing

Doing

array_ops

mindspore.ops.operations.UnsortedSegmentProd

Supported

Doing

Doing

array_ops

mindspore.ops.operations.Concat

Supported

Supported

Doing

array_ops

mindspore.ops.operations.ParallelConcat

Supported

Doing

Doing

array_ops

mindspore.ops.operations.Slice

Supported

Supported

Doing

array_ops

mindspore.ops.operations.Select

Supported

Supported

Doing

array_ops

mindspore.ops.operations.StridedSlice

Supported

Supported

Doing

array_ops

mindspore.ops.operations.Diag

Doing

Doing

Doing

array_ops

mindspore.ops.operations.DiagPart

Doing

Doing

Doing

array_ops

mindspore.ops.operations.Eye

Supported

Supported

Supported

array_ops

mindspore.ops.operations.ScatterNd

Supported

Doing

Doing

array_ops

mindspore.ops.operations.ResizeNearestNeighbor

Supported

Doing

Doing

array_ops

mindspore.ops.operations.GatherNd

Supported

Doing

Doing

array_ops

mindspore.ops.operations.ApplyFtrl

Supported

Supported

Doing

array_ops

mindspore.ops.operations.SparseApplyFtrl

Supported

Doing

Doing

array_ops

mindspore.ops.operations.FusedSparseFtrl

Doing

Doing

Supported

array_ops

mindspore.ops.operations.SparseApplyFtrlV2

Supported

Doing

Doing

array_ops

mindspore.ops.operations.ScatterNdUpdate

Supported

Doing

Doing

array_ops

mindspore.ops.operations.ScatterUpdate

Supported

Doing

Doing

array_ops

mindspore.ops.operations.ScatterMul

Supported

Doing

Doing

array_ops

mindspore.ops.operations.ScatterDiv

Supported

Doing

Doing

array_ops

mindspore.ops.operations.SpaceToDepth

Supported

Doing

Doing

array_ops

mindspore.ops.operations.DepthToSpace

Supported

Doing

Doing

array_ops

mindspore.ops.operations.SpaceToBatch

Supported

Doing

Doing

array_ops

mindspore.ops.operations.SpaceToBatchND

Supported

Doing

Doing

array_ops

mindspore.ops.operations.BatchToSpace

Supported

Doing

Doing

array_ops

mindspore.ops.operations.BatchToSpaceND

Supported

Doing

Doing

array_ops

mindspore.ops.operations.IsFinite

Supported

Supported

Doing

array_ops

mindspore.ops.operations.InplaceUpdate

Supported

Doing

Doing

array_ops

mindspore.ops.operations.ScatterSub

Supported

Doing

Doing

array_ops

mindspore.ops.operations.ScatterMax

Supported

Doing

Doing

array_ops

mindspore.ops.operations.ScatterMin

Supported

Doing

Doing

array_ops

mindspore.ops.operations.ScatterNdAdd

Supported

Doing

Doing

array_ops

mindspore.ops.operations.ScatterNdSub

Supported

Doing

Doing

array_ops

mindspore.ops.operations.ScatterNonAliasingAdd

Supported

Doing

Doing

array_ops

mindspore.ops.operations.Rint

Supported

Doing

Doing

array_ops

mindspore.ops.operations.ReverseV2

Supported

Doing

Doing

array_ops

mindspore.ops.operations.ReduceOp

Supported

Supported

Doing

comm_ops

mindspore.ops.operations.AllReduce

Supported

Supported

Doing

comm_ops

mindspore.ops.operations.AllGather

Supported

Supported

Doing

comm_ops

mindspore.ops.operations.ReduceScatter

Doing

Supported

Doing

comm_ops

mindspore.ops.operations.Broadcast

Supported

Doing

Doing

comm_ops

mindspore.ops.operations.ControlDepend

Supported

Supported

Supported

control_ops

mindspore.ops.operations.GeSwitch

Doing

Doing

Doing

control_ops

mindspore.ops.operations.Merge

Doing

Doing

Doing

control_ops

mindspore.ops.operations.ScalarSummary

Supported

Supported

Supported

debug_ops

mindspore.ops.operations.ImageSummary

Supported

Supported

Supported

debug_ops

mindspore.ops.operations.TensorSummary

Supported

Supported

Supported

debug_ops

mindspore.ops.operations.HistogramSummary

Supported

Supported

Supported

debug_ops

mindspore.ops.operations.InsertGradientOf

Supported

Supported

Supported

debug_ops

mindspore.ops.operations.Print

Supported

Doing

Doing

debug_ops

mindspore.ops.operations.Assign

Supported

Supported

Doing

other_ops

mindspore.ops.operations.BoundingBoxEncode

Supported

Doing

Doing

other_ops

mindspore.ops.operations.BoundingBoxDecode

Supported

Doing

Doing

other_ops

mindspore.ops.operations.PopulationCount

Supported

Doing

Doing

other_ops

mindspore.ops.operations.CheckValid

Supported

Doing

Doing

other_ops

mindspore.ops.operations.IOU

Supported

Doing

Doing

other_ops

mindspore.ops.operations.MakeRefKey

Supported

Supported

Supported

other_ops

mindspore.ops.operations.InTopK

Supported

Doing

Doing

other_ops

mindspore.ops.operations.StandardNormal

Supported

Supported

Doing

random_ops

mindspore.ops.operations.Gamma

Supported

Doing

Doing

random_ops

mindspore.ops.operations.Poisson

Supported

Doing

Doing

random_ops

mindspore.ops.operations.UniformInt

Supported

Doing

Doing

random_ops

mindspore.ops.operations.UniformReal

Supported

Doing

Doing

random_ops

mindspore.ops.operations.RandomChoiceWithMask

Doing

Doing

Doing

random_ops

mindspore.ops.operations.RandomCategorical

Supported

Doing

Doing

random_ops

mindspore.ops.operations.ScalarCast

Supported

Supported

Supported

inner_ops

mindspore.ops.operations.ReverseSequence

Supported

Doing

Doing

array_ops

mindspore.ops.operations.CropAndResize

Supported

Doing

Doing

image_ops

Implicit Type Conversion

conversion rules

  • Scalar and Tensor operations: during operation, the scalar is automatically converted to Tensor, and the data type is consistent with the Tensor data type involved in the operation; when Tensor is a bool data type and the scalar is int or float, both the scalar and Tensor are converted to the Tensor with the data type of int32 or float32.

  • Tensor operation of different data types: the priority of data type is bool < uint8 < int8 < int16 < int32 < int64 < float16 < float32 <float64, during the operation, first determine the data type with the relatively highest priority among the Tensors involved in the operation, and then convert the low priority data type Tensor to the relatively highest priority data type; when the Tensor of int8 and uint8 data types are operated, they are converted to int16 Tensor.

  • Data type conversion of Parameter is not supported: If inferred according to the conversion rules, RuntimeError exception will be thrown when the data type conversion of Parameter defined in the network is required.

data types involved in conversion

  • bool

  • int8

  • uint8

  • int16

  • int32

  • int64

  • float16

  • float32

  • float64

support ops

op name

mindspore.ops.operations.Assign

mindspore.ops.operations.AssignSub

mindspore.ops.operations.ApplyMomentum

mindspore.ops.operations.FusedSparseAdam

mindspore.ops.operations.FusedSparseLazyAdam

mindspore.ops.operations.FusedSparseFtrl

mindspore.ops.operations.FusedSparseProximalAdagrad

mindspore.ops.operations.ApplyAdaMax

mindspore.ops.operations.ApplyAdadelta

mindspore.ops.operations.ApplyAdagrad

mindspore.ops.operations.ApplyAdagradV2

mindspore.ops.operations.SparseApplyAdagrad

mindspore.ops.operations.SparseApplyAdagradV2

mindspore.ops.operations.ApplyProximalAdagrad

mindspore.ops.operations.SparseApplyProximalAdagrad

mindspore.ops.operations.ApplyAddSign

mindspore.ops.operations.ApplyPowerSign

mindspore.ops.operations.ApplyGradientDescent

mindspore.ops.operations.ApplyProximalGradientDescent

mindspore.ops.operations.SparseApplyFtrl

mindspore.ops.operations.SparseApplyFtrlV2

mindspore.ops.operations.BitwiseAnd

mindspore.ops.operations.BitwiseOr

mindspore.ops.operations.BitwiseXor

mindspore.ops.operations.TensorAdd

mindspore.ops.operations.Sub

mindspore.ops.operations.Mul

mindspore.ops.operations.Pow

mindspore.ops.operations.Minimum

mindspore.ops.operations.Maximum

mindspore.ops.operations.RealDiv

mindspore.ops.operations.Div

mindspore.ops.operations.DivNoNan

mindspore.ops.operations.FloorDiv

mindspore.ops.operations.TruncateDiv

mindspore.ops.operations.TruncateMod

mindspore.ops.operations.Mod

mindspore.ops.operations.FloorMod

mindspore.ops.operations.Atan2

mindspore.ops.operations.Equal

mindspore.ops.operations.ApproximateEqual

mindspore.ops.operations.NotEqual

mindspore.ops.operations.Greater

mindspore.ops.operations.GreaterEqual

mindspore.ops.operations.Less

mindspore.ops.operations.LessEqual

mindspore.ops.operations.LogicalAnd

mindspore.ops.operations.LogicalOr

mindspore.ops.operations.ScatterNdUpdate

mindspore.ops.operations.ScatterNdAdd

mindspore.ops.operations.ScatterNdSub

mindspore.ops.operations.ScatterNonAliasingAdd

mindspore.ops.operations.ScatterUpdate

mindspore.ops.operations.ScatterMax

mindspore.ops.operations.ScatterMin

mindspore.ops.operations.ScatterAdd

mindspore.ops.operations.ScatterSub

mindspore.ops.operations.ScatterMul

mindspore.ops.operations.ScatterDiv