Operator Lists
The current version of Auto Tune supports only the auto tuning of AI Core operators whose computation logic is implemented using DSL APIs. This section lists the operators that support RL and GA tuning.
Operators Supporting RL Tuning
- Abs
- AbsGrad
- AcosGrad
- Add
- AsinGrad
- AsinhGrad
- AtanGrad
- BiasAddGrad
- BNTrainingUpdate
- Ceil
- ConcatD
- Cos
- Cosh
- Div
- 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
- ReduceAllD
- ReduceAnyD
- ReduceMaxD
- ReduceMeanD
- ReduceSumD
- Relu
- ReluGrad
- Relu6
- Relu6Grad
- Rint
- Round
- Rsqrt
- RsqrtGrad
- Selu
- Sigmoid
- SigmoidCrossEntropyWithLogits
- SigmoidGrad
- Sign
- Sinh
- SmoothL1Loss
- SoftmaxV2
- SoftmaxGrad
- Softplus
- Softsign
- SplitD
- Sqrt
- SqrtGrad
- Square
- StridedSliceD
- Sub
- Tanh
- TanhGrad
- SoftmaxCrossEntropyWithLogits
- MatMul
- GNTrainingReduce
- GNTrainingUpdate
- CosineEmbeddingLoss
- INTrainingReduceV2
- INTrainingUpdateV2
The preceding list summarizes all operators supporting RL tuning, but might vary according to the Ascend AI Processor version. For details, see CANN Operator List (Inference).
Operators Supporting GA Tuning
- AvgPool
- Conv2D
- Conv2DBackpropInput
- Conv2DBackpropFilter
- Conv2DCompress
- Conv2DTranspose
- Conv3D
- Conv3DBackpropInput
- Conv3DBackpropFilter
- DepthwiseConv2D
- DepthwiseConv2DBackpropInput
- DepthwiseConv2DBackpropFilter
- Deconvolution
- GEMM
- Pooling
The preceding list summarizes all operators supporting GA tuning, but might vary according to the Ascend AI Processor version. For details, see CANN Operator List (Inference).