算子列表
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
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
父主题: 附录