算子列表
Abs
AccumulateNV2
Add
AddN
Adds
BatchNorm3D
BiasAdd
BiasAddGrad
BNInfer
BNTrainingReduce
BNTrainingReduceGrad
BNTrainingUpdate
BNTrainingUpdateGrad
BNTrainingUpdateV2
BNTrainingUpdateV3(仅
Atlas 200/300/500 推理产品
支持)
ClipByNormNoDivSum
ClipByValue
DiagPartD
Div
DivNoNan
DynamicRNN
Elu
EluGrad
Exp
ExpandD
Floor
FloorDiv
Gelu
InstanceNorm
L2Loss
LambUpdateWithLrV2
LayerNormBetaGammaBackpropV2(仅
Atlas 200/300/500 推理产品
支持)
LeakyRelu
Log
Log1p
LpNorm(仅
Atlas 200/300/500 推理产品
支持)
Maximum
MaxPool
MaxPool3D
MaxPoolV3
Minimum
Mul
Muls
Neg
Pow
PRelu
RealDiv
ReduceAllD
ReduceMaxD
ReduceMeanD
ReduceMinD
ReduceSumD
Relu
ReluV2
Round
Rsqrt
RsqrtGrad
Sigmoid
Sign
SoftmaxCrossEntropyWithLogits
Softplus
SplitVD(仅
Atlas 200/300/500 推理产品
支持)
SqrtGrad
Square
SquaredDifference
SquareSumV1
SquareSumV2
StridedSliceD
Sub
Tanh
TileD
Unpack
ReluGradV2
AvgPool
AvgPoolV2
AvgPool3D
BatchMatMulV3
Deconvolution
Conv1D
Conv2D
Conv2DCompress
Conv2DBackpropInput
Conv2DBackpropFilter
DeformableConv2D
GEMM
MatMul
MatMulV2
MatMulV3
MatMulV2Compress
BatchMatMul
BatchMatMulV2
FullyConnection
FullyConnectionCompress
DepthwiseConv2D
DepthwiseConv2DBackpropInput
DepthwiseConv2DBackpropFilter
Conv3D
Conv3DBackpropInput
Conv3DBackpropFilter
AvgPool3DGrad
Pooling
Conv2DTranspose
Conv3DTranspose
GatherV2(仅支持
Atlas 推理系列产品
和
Atlas 训练系列产品
)
以上算子列表为支持的全量算子,由于
昇腾AI处理器
型号不同,支持的范围会有差异,上述列表中的算子在
昇腾AI处理器
中是否支持请以
《
算子清单
》
为准。
父主题:
附录