Operator List

mindspore.nn

Operation Ascend GPU CPU Operator Type
mindspore.nn.Softmax Supported Supported Supported layer/activation
mindspore.nn.LogSoftmax Supported Doing Doing layer/activation
mindspore.nn.ReLU Supported Supported Supported layer/activation
mindspore.nn.ReLU6 Supported Supported Doing layer/activation
mindspore.nn.HSwish Supported Supported Doing layer/activation
mindspore.nn.HSigmoid Supported Supported Doing layer/activation
mindspore.nn.LeakyReLU Supported Doing Doing layer/activation
mindspore.nn.Tanh Supported Doing Doing layer/activation
mindspore.nn.GELU Supported Doing Doing layer/activation
mindspore.nn.Sigmoid Supported Doing Doing layer/activation
mindspore.nn.PReLU Supported Doing Doing layer/activation
mindspore.nn.Dropout Supported Doing Doing layer/basic
mindspore.nn.Flatten Supported Supported Doing layer/basic
mindspore.nn.Dense Supported Supported Doing layer/basic
mindspore.nn.ClipByNorm Doing Doing Doing layer/basic
mindspore.nn.Norm Doing Doing Doing layer/basic
mindspore.nn.OneHot Supported Supported Doing layer/basic
mindspore.nn.SequentialCell Doing 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.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.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 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.Adam Supported Doing Doing optim/adam
mindspore.nn.AdamWeightDecay Supported Doing Doing optim/adam
mindspore.nn.AdamWeightDecayDynamicLR Supported Doing Doing optim/adam
mindspore.nn.Lamb Supported Doing Doing optim/lamb
mindspore.nn.LARS Doing Doing Doing optim/lars
mindspore.nn.Momentum Supported Supported Doing optim/momentum
mindspore.nn.Optimizer Supported Supported Doing optim/optimizer
mindspore.nn.RMSProp Supported Doing Doing optim/optimizer
mindspore.nn.SGD Supported Doing Doing optim/sgd
mindspore.nn.WithLossCell Supported Supported Doing wrap/cell_wrapper
mindspore.nn.WithGradCell Supported Doing Doing wrap/cell_wrapper
mindspore.nn.TrainOneStepCell Supported Supported Doing wrap/cell_wrapper
mindspore.nn.DataWrapper Doing Doing 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 Doing Doing Doing nn_ops
mindspore.ops.operations.LogSoftmax Supported Doing Doing nn_ops
mindspore.ops.operations.Softplus Supported Doing Doing nn_ops
mindspore.ops.operations.ReLU Supported Supported Supported nn_ops
mindspore.ops.operations.ReLUV2 Supported Doing Doing nn_ops
mindspore.ops.operations.ReLU6 Supported Supported Doing nn_ops
mindspore.ops.operations.HSwish Supported Supported Doing nn_ops
mindspore.ops.operations.HSigmoid Supported Supported Doing nn_ops
mindspore.ops.operations.Sigmoid Supported Doing Doing nn_ops
mindspore.ops.operations.Tanh Supported Doing Doing nn_ops
mindspore.ops.operations.BatchNorm 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.ApplyRMSProp Supported Doing Doing 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 Doing 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 Doing 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.SigmoidCrossEntropyWithLogits Supported Doing 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 Doing Doing nn_ops
mindspore.ops.operations.BinaryCrossEntropy Supported Doing Doing nn_ops
mindspore.ops.operations.LARSUpdate 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 Doing 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 Doing Doing math_ops
mindspore.ops.operations.CumSum Doing 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 Doing Doing math_ops
mindspore.ops.operations.Mul Supported Supported Supported math_ops
mindspore.ops.operations.Square Supported Doing Doing math_ops
mindspore.ops.operations.SquareSumAll Supported Doing Doing math_ops
mindspore.ops.operations.Rsqrt Supported Doing Doing math_ops
mindspore.ops.operations.Sqrt Supported Doing Doing math_ops
mindspore.ops.operations.Reciprocal Supported Supported Doing math_ops
mindspore.ops.operations.Pow Supported Doing 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 Doing Doing math_ops
mindspore.ops.operations.Maximum Supported Doing Doing math_ops
mindspore.ops.operations.RealDiv Supported Supported Doing math_ops
mindspore.ops.operations.Div 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 Doing Doing math_ops
mindspore.ops.operations.Greater Supported Doing Doing math_ops
mindspore.ops.operations.GreaterEqual Supported Doing Doing math_ops
mindspore.ops.operations.Less Supported Doing Doing math_ops
mindspore.ops.operations.Atan2 Doing Doing Doing math_ops
mindspore.ops.operations.LessEqual Supported Doing Doing math_ops
mindspore.ops.operations.LogicalNot Supported Doing Doing math_ops
mindspore.ops.operations.LogicalAnd Supported Doing Doing math_ops
mindspore.ops.operations.LogicalOr 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.Acos Doing 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.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.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 Doing 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 Doing Doing Doing array_ops
mindspore.ops.operations.ArgMaxWithValue Supported Doing 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 Doing Doing array_ops
mindspore.ops.operations.UnsortedSegmentMin Doing Doing Doing array_ops
mindspore.ops.operations.Concat Supported Supported Doing array_ops
mindspore.ops.operations.Slice Supported Supported Doing array_ops
mindspore.ops.operations.Select Supported Doing 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 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.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.BatchToSpace Supported Doing Doing array_ops
mindspore.ops.operations.IsFinite Supported Doing Doing array_ops
mindspore.ops.operations.ReduceOp Supported Supported Doing comm_ops
mindspore.ops.operations.AllReduce Doing Supported Doing comm_ops
mindspore.ops.operations.AllGather Doing Supported Doing comm_ops
mindspore.ops.operations.ReduceScatter Doing Supported Doing comm_ops
mindspore.ops.operations.Broadcast Doing 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 Doing debug_ops
mindspore.ops.operations.ImageSummary Supported Supported Doing debug_ops
mindspore.ops.operations.TensorSummary Supported Supported Doing debug_ops
mindspore.ops.operations.HistogramSummary Supported Supported Doing 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 Doing Doing other_ops
mindspore.ops.operations.BoundingBoxEncode Supported Doing Doing other_ops
mindspore.ops.operations.BoundingBoxDecode 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.RandomChoiceWithMask Doing Doing Doing random_ops
mindspore.ops.operations.ScalarCast Supported Supported Supported inner_ops