算子支持

mindspore.nn

操作名 Ascend GPU CPU 算子类别
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 Doing 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 Doing Doing layer/activation
mindspore.nn.GELU Supported Doing Doing layer/activation
mindspore.nn.Sigmoid Supported Doing Doing layer/activation
mindspore.nn.PReLU Doing Doing Doing layer/activation
mindspore.nn.Dropout Doing Doing Doing layer/basic
mindspore.nn.Flatten Doing 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 Doing layer/lstm
mindspore.nn.BatchNorm1d Supported Doing Doing layer/normalization
mindspore.nn.BatchNorm2d Supported Supported 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 Doing Doing Doing loss/loss
mindspore.nn.SoftmaxCrossEntropyWithLogits Supported Supported Supported loss/loss
mindspore.nn.SoftmaxCrossEntropyExpand Supported Doing Doing loss/loss
mindspore.nn.Adam Doing 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 Supported optim/momentum
mindspore.nn.Optimizer Supported Supported Doing optim/optimizer
mindspore.nn.SGD Doing 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

操作名 Ascend GPU CPU 算子类别
mindspore.ops.operations.Flatten Doing 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 Doing Doing Doing nn_ops
mindspore.ops.operations.RMSprop Doing Doing Doing nn_ops
mindspore.ops.operations.ExtractImagePatches Doing Doing Doing nn_ops
mindspore.ops.operations.MirrorPad Doing Doing Doing nn_ops
mindspore.ops.operations.MirrorPadGrad Doing Doing Doing nn_ops
mindspore.ops.operations.unpack Doing Doing Doing nn_ops
mindspore.ops.operations.pack Doing Doing Doing nn_ops
mindspore.ops.operations.L2Loss Supported Doing Doing nn_ops
mindspore.ops.operations.LogSoftmax 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.ReLU6Grad 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 Doing Doing nn_ops
mindspore.ops.operations.Tanh Supported Doing Doing nn_ops
mindspore.ops.operations.FusedBatchNorm Doing Supported 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.MaxPoolWithArgmaxGrad Supported Doing Doing nn_ops
mindspore.ops.operations.MaxPool Supported Supported Supported nn_ops
mindspore.ops.operations.AvgPool Doing Supported Doing nn_ops
mindspore.ops.operations.Conv2DBackpropInput Doing 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.SmoothL1Loss Doing Doing Doing nn_ops
mindspore.ops.operations.SGD Doing Doing Doing nn_ops
mindspore.ops.operations.LayerNorm Supported Doing Doing nn_ops
mindspore.ops.operations.L2Normalize Doing 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 Doing 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 Doing Doing Doing nn_ops
mindspore.ops.operations.LSTM Doing Supported Doing nn_ops
mindspore.ops.operations.SigmoidCrossEntropyWithLogits Supported Doing Doing nn_ops
mindspore.ops.operations.Pad Supported Doing Doing nn_ops
mindspore.ops.operations.ROIAlign Doing Doing Doing nn_ops
mindspore.ops.operations.Adam Supported Doing Doing nn_ops
mindspore.ops.operations.BinaryCrossEntropy Doing Doing Doing nn_ops
mindspore.ops.operations.SparseApplyAdagrad Doing Doing Doing nn_ops
mindspore.ops.operations.LARSUpdate Doing 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 Doing Doing Doing math_ops
mindspore.ops.operations.CumProd Doing 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.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.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 Doing 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 Supported 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.Cos Doing 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.Sin Doing Doing Doing math_ops
mindspore.ops.operations.NMSWithMask Doing Doing Doing math_ops
mindspore.ops.operations.Abs Supported Doing Doing math_ops
mindspore.ops.operations.AbsGrad 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 Doing 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 Doing 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 Doing 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.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 Doing Doing Doing array_ops
mindspore.ops.operations.ScatterNdUpdate Supported Doing Doing array_ops
mindspore.ops.operations.SpaceToDepth Doing Doing Doing array_ops
mindspore.ops.operations.DepthToSpace Doing 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 Doing Doing Doing other_ops
mindspore.ops.operations.BoundingBoxDecode Doing Doing Doing other_ops
mindspore.ops.operations.CheckValid Doing Doing Doing other_ops
mindspore.ops.operations.IOU Doing 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