CANN算子清单
CANN算子库支持的算子规格说明,如下所示:
算子规格说明 (Atlas 200&300&500 推理产品)
算子规格说明 (Atlas 推理系列产品,Ascend 310P处理器)
- 以下算子涉及且已支持确定性计算:"BatchMatMulV2", "BiasAddGrad", "BinaryCrossEntropy", "BN3DTrainingReduce", "BN3DTrainingUpdateGrad", "BNTrainingReduce", "BNTrainingUpdateGrad", "Conv2DBackpropFilter", "FusedMulAddNL2loss", "KLDiv", "LayerNormBetaGammaBackpropV2", "MseLoss", "MatMul", "MatMulV2", "NLLLoss", "ReduceMean", ReduceMeanD", "ReduceSum", "ReduceSumD", ""ScatterAdd", "ScatterNd", "ScatterNdAdd", "SquareSumV1", "UnsortedSegmentSum", "EmbeddingDenseGrad", "GroupNormGrad", "IndexPut", "IndexPutV2", "LpNormReduceV2", "MseLoss", "ScatterElements", "LpNormV2", "InplaceIndexAdd", "Histogram"。
- "resizegradD"算子涉及但未支持确定性计算。