aclnnAddLayerNorm

产品支持情况

[object Object]undefined

功能说明

• 接口功能：实现AddLayerNorm功能。
• 计算公式：$$x = x1 + x2 + biasOptional$$ $$rstd = {{1}\over\sqrt {Var(x)+eps}}$$ $$y = (x-E(x)) * rstd * gamma + beta$$ 其中，E(x)表示均值，Var(x)表示方差，均需要在算子内部计算得到。

函数原型

每个算子分为，必须先调用[object Object]接口获取入参并根据计算流程所需workspace大小，再调用[object Object]接口执行计算。

aclnnAddLayerNormGetWorkspaceSize

• 参数说明：

  [object Object]
  • [object Object]Atlas 推理系列产品[object Object]：
    • 参数[object Object]、[object Object]、[object Object]、[object Object]、[object Object]、[object Object]、[object Object]的数据类型不支持BFLOAT16。
    • 参数[object Object]、[object Object]在当前产品使用场景下无效。

• 返回值：

  aclnnStatus：返回状态码，具体参见。

  第一段接口完成入参校验，出现以下场景时报错：

  [object Object]

aclnnAddLayerNorm

• 参数说明：

  [object Object]

• 返回值：

  aclnnStatus：返回状态码，具体参见。

约束说明

• 功能维度
  • 数据类型支持
    • [object Object]Atlas 推理系列产品[object Object]：x1、x2、beta、gamma、biasOptional支持FLOAT32、FLOAT16。
    • [object Object]Atlas A2 训练系列产品/Atlas A2 推理系列产品[object Object]、[object Object]Atlas A3 训练系列产品/Atlas A3 推理系列产品[object Object]：x1、x2、beta、gamma、biasOptional支持FLOAT32、FLOAT16、BFLOAT16。
    • rstdOut、meanOut支持：FLOAT32。
  • 数据格式支持：ND。
  • [object Object]Atlas 推理系列产品[object Object]：x1、x2、beta、gamma、biasOptional五个输入的尾轴长度必须大于等于32Bytes。
• 未支持类型说明
  • DOUBLE：不支持DOUBLE。
• 边界值场景说明
  • 当输入是Inf时，输出为Inf。
  • 当输入是NaN时，输出为NaN。
• 各产品支持数据类型说明
  • [object Object]Atlas A2 训练系列产品/Atlas A2 推理系列产品[object Object]、[object Object]Atlas A3 训练系列产品/Atlas A3 推理系列产品[object Object]：[object Object]undefined
  • [object Object]Atlas 推理系列产品[object Object]：[object Object]undefined
• 确定性计算：
  • aclnnAddLayerNorm默认确定性实现。

调用示例

示例代码如下，仅供参考，具体编译和执行过程请参考。