下载
EN
注册
我要评分
文档获取效率
文档正确性
内容完整性
文档易理解
在线提单
论坛求助
昇腾小AI

接口列表

算子归属领域

算子接口可被归属为如下领域:

  • aclnn_ops_infer:NN网络算子推理库
  • aclnn_ops_train:NN网络算子训练库
  • aclnn_math:数学算子库
  • aclnn_rand:随机数算子库

其中推理库依赖数学库,训练库依赖推理库、数学库与随机库。

每个算子接口所属领域如下表所示。

aclnn api aclnn_ops_infer aclnn_ops_train aclnn_math aclnn_rand
aclnnAddLayerNorm
aclnnAddLayerNormGrad
aclnnAddRmsNorm
aclnnAllGatherMatmul
aclnnAlltoAllAllGatherBatchMatMul
aclnnApplyRotaryPosEmb
aclnnBidirectionLSTM
aclnnBatchMatmulReduceScatterAlltoAll
aclnnDeformableConv2dBackward
aclnnDeepNorm
aclnnDeepNormGrad
aclnnDynamicQuant
aclnnEinsum
aclnnFFN
aclnnFFNV2
aclnnFFNV3
aclnnFlashAttentionScore
aclnnFlashAttentionScoreGrad
aclnnFlashAttentionScoreV2
aclnnFlashAttentionScoreGradV2
aclnnFlashAttentionVarLenScore
aclnnFlashAttentionVarLenScoreV2
aclnnFlashAttentionUnpaddingScoreGrad
aclnnFlashAttentionUnpaddingScoreGradV2
aclnnFusedInferAttentionScore
aclnnFusedInferAttentionScoreV2
aclnnGroupedMatmul
aclnnGroupedMatMulAllReduce
aclnnGroupedMatmulV2
aclnnGroupedMatmulV3
aclnnGroupedMatmulV4
aclnnIncreFlashAttention
aclnnIncreFlashAttentionV2
aclnnIncreFlashAttentionV3
aclnnIncreFlashAttentionV4
aclnnInplaceQuantScatter
aclnnInstanceNorm
aclnnMatmulAllReduce
aclnnMatmulAllReduceAddRmsNorm
aclnnMatmulAllReduceV2
aclnnMatmulReduceScatter
aclnnMoeFinalizeRouting
aclnnMoeGatingTopKSoftmax
aclnnMoeInitRouting
aclnnMoeTokenPermute
aclnnMoeTokenPermuteGrad
aclnnMoeTokenUnpermute
aclnnMoeTokenUnpermuteGrad
aclnnMrgbaCustom
aclnnMultiScaleDeformableAttentionGrad
aclnnPromptFlashAttention
aclnnPromptFlashAttentionV2
aclnnPromptFlashAttentionV3
aclnnQuantMatmulAllReduce
aclnnQuantMatmulAllReduceV2
aclnnQuantMatmulAllReduceV3
aclnnQuantMatmulAllReduceAddRmsNorm
aclnnRmsNorm
aclnnRmsNormGrad
aclnnRingAttentionUpdate
aclnnSwinTransformerLnQkvQuant
aclnnSwinAttentionScoreQuant
aclnnStressDetect
aclnnStressDetectWithPressure
aclnnStressDetectRecover
aclnnWeightQuantBatchMatmul
aclnnWeightQuantBatchMatmulV2
aclnnWeightQuantMatmulAllReduce
aclnnWeightQuantMatmulAllReduceAddRmsNorm
aclRfft1D

废弃接口

废弃接口 说明
aclnnWeightQuantBatchMatmul 此接口后续版本会废弃,请使用 aclnnWeightQuantBatchMatmulV2接口。
搜索结果
找到“0”个结果

当前产品无相关内容

未找到相关内容,请尝试其他搜索词