昇腾社区首页
中文
注册

CANN算子清单

CANN算子库支持的算子规格说明,如下所示:

  • 以下算子涉及且已支持确定性计算:"BatchMatMulV2", "BiasAddGrad", "BinaryCrossEntropy", "BN3DTrainingReduce", "BN3DTrainingUpdateGrad", "BNTrainingReduce", "BNTrainingUpdateGrad", "Conv2DBackpropFilter", "KLDiv", "MseLoss", "MatMul", "MatMulV2", "NLLLoss", "ReduceMean", "ReduceSum", "ScatterAdd", "ScatterNd", "ScatterNdAdd", "UnsortedSegmentSum"。
  • "groupnormgrad"算子涉及但未支持确定性计算。