昇腾社区首页
中文
注册

算子列表

Auto Tune工具当前版本仅支持使用DSL接口实现计算逻辑的AI Core算子的自动调优,本章节介绍RL调优模式与GA调优模式支持的算子列表。

RL调优模式支持的算子

  • Abs
  • AbsGrad
  • AcosGrad
  • Add
  • AsinGrad
  • AsinhGrad
  • AtanGrad
  • BiasAddGrad
  • BNTrainingUpdate
  • Cast
  • Ceil
  • Concat
  • Cos
  • Cosh
  • Div
  • DynamicRNN
  • Elu
  • EluGrad
  • Equal
  • Erf
  • Erfc
  • Exp
  • Expm1
  • Floor
  • Gelu
  • GeluGrad
  • Inv
  • InvGrad
  • L2Loss
  • L2Normalize
  • L2NormalizeGrad
  • Log
  • LogSoftmaxV2
  • LogSoftmaxGrad
  • Log1p
  • Maximum
  • Mod
  • Mul
  • Neg
  • OnesLike
  • Pow
  • PReluGrad
  • Reciprocal
  • ReciprocalGrad
  • ReduceAll
  • ReduceAny
  • ReduceMax
  • ReduceMean
  • ReduceSum
  • Relu
  • ReluGrad
  • Relu6
  • Relu6Grad
  • Rint
  • Round
  • Rsqrt
  • RsqrtGrad
  • Selu
  • Sigmoid
  • SigmoidCrossEntropyWithLogits
  • SigmoidGrad
  • Sign
  • Sinh
  • SmoothL1Loss
  • SoftmaxV2
  • SoftmaxGrad
  • Softplus
  • Softsign
  • Split
  • Sqrt
  • SqrtGrad
  • Square
  • StridedSlice
  • Sub
  • Tanh
  • TanhGrad
  • SoftmaxCrossEntropyWithLogits
  • GNTrainingReduce
  • GNTrainingUpdate
  • CosineEmbeddingLoss
  • INTrainingReduceV2
  • INTrainingUpdateV2

以上算子列表为RL支持的全量算子,由于昇腾AI处理器型号不同,支持的范围会有差异,上述列表中的算子在昇腾AI处理器中是否支持请以算子清单为准。

GA调优模式支持的算子

  • AvgPool
  • Conv2D
  • Conv2DBackpropInput
  • Conv2DBackpropFilter
  • Conv2DCompress
  • Conv3D
  • Conv3DBackpropInput
  • Conv3DBackpropFilter
  • DepthwiseConv2D
  • DepthwiseConv2DBackpropInput
  • DepthwiseConv2DBackpropFilter
  • Deconvolution
  • GEMM
  • MatMul
  • MatMulV2
  • BatchMatMul
  • BatchMatMulV2
  • MatMulV2Compress
  • FullyConnection
  • FullyConnectionCompress
  • AvgPool3DGrad
  • Pooling
  • Conv2DTranspose
  • Conv3DTranspose

    以上算子列表为GA支持的全量算子,由于昇腾AI处理器型号不同,支持的范围会有差异,上述列表中的算子在昇腾AI处理器中是否支持请以算子清单为准。