算子支持
- mindspore.nn
- mindspore.ops.operations

算子支持

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.LeakyReLU	Doing	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	Doing	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	Doing	Supported	Doing	layer/conv
mindspore.nn.Embedding	Doing	Supported	Doing	layer/embedding
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	Doing	Doing	Doing	loss/loss
mindspore.nn.SmoothL1Loss	Doing	Doing	Doing	loss/loss
mindspore.nn.SoftmaxCrossEntropyWithLogits	Supported	Supported	Supported	loss/loss
mindspore.nn.SoftmaxCrossEntropyExpand	Doing	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	Doing	Doing	Doing	optim/lamb
mindspore.nn.LARS	Doing	Doing	Doing	optim/lars
mindspore.nn.Momentum	Supported	Supported	Supported	optim/momentum
mindspore.nn.Optimizer	Doing	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	Doing	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.LogSoftmax	Supported	Doing	Doing	nn_ops
mindspore.ops.operations.ReLU	Supported	Supported	Supported	nn_ops
mindspore.ops.operations.ReLU6	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	Doing	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	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	Doing	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	Doing	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	Doing	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	Doing	Doing	Doing	math_ops
mindspore.ops.operations.Greater	Supported	Doing	Doing	math_ops
mindspore.ops.operations.GreaterEqual	Doing	Doing	Doing	math_ops
mindspore.ops.operations.Less	Supported	Doing	Doing	math_ops
mindspore.ops.operations.LessEqual	Supported	Doing	Doing	math_ops
mindspore.ops.operations.LogicalNot	Doing	Doing	Doing	math_ops
mindspore.ops.operations.LogicalAnd	Doing	Doing	Doing	math_ops
mindspore.ops.operations.LogicalOr	Doing	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.Sin	Doing	Doing	Doing	math_ops
mindspore.ops.operations.NMSWithMask	Doing	Doing	Doing	math_ops
mindspore.ops.operations.Abs	Doing	Doing	Doing	math_ops
mindspore.ops.operations.Sign	Doing	Doing	Doing	math_ops
mindspore.ops.operations.Round	Doing	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	Supported	Supported	array_ops
mindspore.ops.operations.Argmax	Doing	Supported	Supported	array_ops
mindspore.ops.operations.Argmin	Doing	Doing	Doing	array_ops
mindspore.ops.operations.ArgMaxWithValue	Doing	Doing	Doing	array_ops
mindspore.ops.operations.ArgMinWithValue	Supported	Doing	Doing	array_ops
mindspore.ops.operations.Tile	Doing	Doing	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.ConcatOffset	Supported	Supported	Supported	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.Eye	Supported	Supported	Supported	array_ops
mindspore.ops.operations.ScatterNd	Supported	Doing	Doing	array_ops
mindspore.ops.operations.ResizeNearestNeighbor	Doing	Doing	Doing	array_ops
mindspore.ops.operations.GatherNd	Doing	Doing	Doing	array_ops
mindspore.ops.operations.ScatterNdUpdate	Doing	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	Doing	Supported	Doing	debug_ops
mindspore.ops.operations.ImageSummary	Doing	Supported	Doing	debug_ops
mindspore.ops.operations.TensorSummary	Doing	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