aclnnGroupNormSwishGrad

产品支持情况

产品	是否支持
[object Object]Atlas A3 训练系列产品/Atlas A3 推理系列产品[object Object]	√
[object Object]Atlas A2 训练系列产品/Atlas 800I A2 推理产品/A200I A2 Box 异构组件[object Object]	√
[object Object]Atlas 200I/500 A2 推理产品[object Object]	×
[object Object]Atlas 推理系列产品 [object Object]	×
[object Object]Atlas 训练系列产品[object Object]	×

功能说明

算子功能：undefined的反向操作。

函数原型

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

• aclnnStatus aclnnGroupNormSwishGradGetWorkspaceSize(const aclTensor *dy, const aclTensor *mean, const aclTensor *rstd, const aclTensor *x, const aclTensor *gamma, const aclTensor *beta, int64_t numGroups, char *dataFormatOptional, double swishScale, bool dgammaIsRequire, bool dbetaIsRequire, const aclTensor *dxOut, const aclTensor *dgammaOut, const aclTensor *dbetaOut, uint64_t *workspaceSize, aclOpExecutor **executor)
• aclnnStatus aclnnGroupNormSwishGrad(void *workspace, uint64_t workspaceSize, aclOpExecutor *executor, aclrtStream stream)

aclnnGroupNormSwishGradGetWorkspaceSize

• 参数说明：

  • dy (aclTensor*, 计算输入)：输入张量，Device侧的aclTensor，反向计算的梯度，维度需大于一维，元素个数需要等于N*C*HxW，数据类型支持FLOAT32、FLOAT16、BFLOAT16，undefined支持ND，支持undefined。

  • mean (aclTensor*, 计算输入)：输入张量，Device侧的aclTensor，正向计算的第二个输出，表示input分组后每个组的均值，元素个数需要等于N*group，数据类型支持FLOAT32、FLOAT16、BFLOAT16，数据类型与$gamma$相同，其中N与$dy$的第0维度保持一致，undefined支持ND，支持undefined。

  • rstd (aclTensor*, 计算输入)：输入张量，Device侧的aclTensor，正向计算的第三个输出，表示input分组后每个组的标准差倒数，元素个数需要等于N*group，数据类型支持FLOAT32、FLOAT16、BFLOAT16，数据类型与$gamma$相同，其中N与$dy$的第0维度保持一致，undefined支持ND，支持undefined。

  • x (aclTensor*, 计算输入)：输入张量，Device侧的aclTensor，正向的输入$x$，维度需大于一维，数据类型支持FLOAT32、FLOAT16、BFLOAT16，undefined支持ND，支持undefined。

  • gamma (aclTensor*, 计算输入)：输入张量，Device侧的aclTensor，表示每个channel的缩放系数，维度为一维，元素个数需要等于C，数据类型支持FLOAT32、FLOAT16、BFLOAT16，undefined支持ND，支持undefined。

  • beta (aclTensor*, 计算输入)：输入张量，Device侧的aclTensor，表示每个channel的偏移系数，维度为一维，元素个数需要等于C，数据类型支持FLOAT32、FLOAT16、BFLOAT16，数据类型与$gamma$相同，undefined支持ND，支持undefined。

  • numGroups (int64_t, 计算输入)：INT64常量，表示将输入gradOut的C维度分为group组，group需大于0，C必须可以被group整除并且比值不能超过4000。

  • dataFormatOptional (char*, 计算输入)：表示数据格式，建议值NCHW。

  • swishScale (double, 计算输入)：Swish计算公式中的系数，建议值1.0。

  • dgammaIsRequire (bool, 计算输入)：是否需要输出dgamma，建议值true。

  • dbetaIsRequire (bool, 计算输入)：是否需要输出dbeta，建议值true。

  • dxOut (aclTensor*, 计算输出)：输出Tensor，Device侧的aclTensor，x的梯度，数据类型支持BFLOAT16、FLOAT16、FLOAT，数据类型和shape与$x$相同，undefined支持ND，支持undefined。

  • dgammaOut (aclTensor*, 计算输出)：输出Tensor，Device侧的aclTensor，gamma的梯度，数据类型支持BFLOAT16、FLOAT16、FLOAT，数据类型和shape与$gamma$相同，undefined支持ND，支持undefined。

  • dbetaOut (aclTensor*, 计算输出)：输出Tensor，Device侧的aclTensor，beta的梯度，数据类型支持BFLOAT16、FLOAT16、FLOAT，数据类型和shape与$gamma$相同，undefined支持ND，支持undefined。

  • workspaceSize (uint64_t*, 出参)：返回需要在Device侧申请的workspace大小。

  • executor (aclOpExecutor**, 出参)：返回op执行器，包含算子计算流程。

• 返回值：

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

aclnnGroupNormSwishGrad

• 参数说明：

  • workspace(void*, 入参)：在Device侧申请的workspace内存地址。
  • workspaceSize(uint64_t, 入参)：在Device侧申请的workspace大小，由第一段接口aclnnGroupNormSwishGradGetWorkspaceSize获取。
  • executor(aclOpExecutor*, 入参)：op执行器，包含了算子计算流程。
  • stream(aclrtStream, 入参)：指定执行任务的Stream。

• 返回值：

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

约束说明

无

调用示例

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