每个算子分为,必须先调用“aclnnGroupNormSwishGradGetWorkspaceSize”接口获取计算所需workspace大小以及包含了算子计算流程的执行器,再调用“aclnnGroupNormSwishGrad”接口执行计算。
[object Object]
[object Object]
确定性计算
- aclnnGroupNormSwishGrad默认非确定性实现,支持通过aclrtCtxSetSysParamOpt开启确定性。
输入shape限制:
- numGroups大于0。
- C能被group整除。
- dy的元素个数等于。
- mean的元素个数等于。
- rstd的元素个数等于。
- x的元素个数等于。
- gamma的元素个数等于C。
- beta的元素个数等于C。
- C与group比值不超过4000。
[object Object]