aclnnDeepNormGrad

产品支持情况

[object Object]undefined

功能说明

• 接口功能：的反向传播，完成张量x、张量gx、张量gamma的梯度计算，以及张量dy的求和计算。

• 计算公式：

  $$dgx_i = tmpone_i * rstd + dvar * tmptwo_i + dmean$$ $$dx_i = alpha * {dgx}_i$$ $$dbeta = \sum_{i=1}^{N} dy_i$$ $$dgamma = \sum_{i=1}^{N} dy_i * rstd * {tmptwo}_i$$

  其中：

  $$oneDiv=-1/SizeOf(gamma)$$ $$tmpone_i = dy_i * gamma$$ $$tmptwo_i = alpha * x_i + {gx}_i - mean$$ $$dvar = (oneDiv) * \sum_{i=1}^{N} {tmpone}_i * {tmptwo}_i * {rstd}^3$$ $$dmean = (oneDiv) * \sum_{i=1}^{N} {tmpone}_i * rstd$$

函数原型

每个算子分为，必须先调用“aclnnDeepNormGradGetWorkspaceSize”接口获取计算所需workspace大小以及包含了算子计算流程的执行器，再调用“aclnnDeepNormGrad”接口执行计算。

aclnnDeepNormGradGetWorkspaceSize

• 参数说明：

  [object Object]
  • [object Object]Atlas 推理系列产品[object Object]：参数[object Object]、[object Object]、[object Object]、[object Object]、[object Object]、[object Object]的数据类型不支持BFLOAT16。

• 返回值：

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

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

  [object Object]

aclnnDeepNormGrad

• 参数说明：

  [object Object]

• 返回值：

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

约束说明

• 未支持类型说明：

  DOUBLE：指令不支持DOUBLE。

• 边界值场景说明：

  • 当输入是Inf时，输出为Inf。
  • 当输入是NaN时，输出为NaN。

• 确定性计算：

  • aclnnDeepNormGrad默认非确定性实现，不支持通过aclrtCtxSetSysParamOpt开启确定性。

调用示例

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