aclnnGroupNormSiluV2
产品支持情况
| 产品 | 是否支持 | 
|---|---|
| [object Object]Atlas A3 训练系列产品/Atlas A3 推理系列产品[object Object] | √ | 
| [object Object]Atlas A2 训练系列产品/Atlas A2 推理系列产品[object Object] | √ | 
| [object Object]Atlas 200I/500 A2 推理产品[object Object] | × | 
| [object Object]Atlas 推理系列产品 [object Object] | √ | 
| [object Object]Atlas 训练系列产品[object Object] | × | 
功能说明
- 接口功能:计算输入self的组归一化结果out,均值meanOut,标准差的倒数rstdOut,以及silu的输出。
 - 计算公式:
- GroupNorm: 记 代表的均值,代表的方差,则
 - Silu: 当activateSilu为True时,会计算Silu, 此时Silu计算公式的x为GroupNorm公式得到的out。
 
 
函数原型
每个算子分为,必须先调用“aclnnGroupNormSiluV2GetWorkspaceSize”接口获取计算所需workspace大小以及包含了算子计算流程的执行器,再调用“aclnnGroupNormSiluV2”接口执行计算。
aclnnStatus aclnnGroupNormSiluV2GetWorkspaceSize(const aclTensor *self, const aclTensor *gamma, const aclTensor *beta, int64_t group, double eps, bool activateSilu, aclTensor *out, aclTensor *meanOut, aclTensor *rstdOut, uint64_t *workspaceSize, aclOpExecutor **executor)aclnnStatus aclnnGroupNormSiluV2(void *workspace, uint64_t workspaceSize, aclOpExecutor *executor, aclrtStream stream)
aclnnGroupNormSiluV2GetWorkspaceSize
参数说明:
- self(aclTensor*, 计算输入):
out计算公式中的,维度需大于一维且小于等于八维,支持ND,支持。- [object Object]Atlas 推理系列产品[object Object]:数据类型支持FLOAT16、FLOAT。
 - [object Object]Atlas A2 训练系列产品/Atlas A2 推理系列产品[object Object]、[object Object]Atlas A3 训练系列产品/Atlas A3 推理系列产品[object Object]:数据类型支持FLOAT16、FLOAT、BFLOAT16。
 
 - gamma(aclTensor*, 计算输入):可选参数,
out计算公式中的,维度为一维,元素数量需与输入的第1维度相同,支持ND,支持。- [object Object]Atlas 推理系列产品[object Object]:数据类型支持FLOAT16、FLOAT。
 - [object Object]Atlas A2 训练系列产品/Atlas A2 推理系列产品[object Object]、[object Object]Atlas A3 训练系列产品/Atlas A3 推理系列产品[object Object]:数据类型支持FLOAT16、FLOAT、BFLOAT16。
 
 - beta(aclTensor*, 计算输入):可选参数,
out计算公式中的,维度为一维,元素数量需与输入的第1维度相同,支持ND,支持。- [object Object]Atlas 推理系列产品[object Object]:数据类型支持FLOAT16、FLOAT。
 - [object Object]Atlas A2 训练系列产品/Atlas A2 推理系列产品[object Object]、[object Object]Atlas A3 训练系列产品/Atlas A3 推理系列产品[object Object]:数据类型支持FLOAT16、FLOAT、BFLOAT16。
 
 - group(int, 计算输入): INT32或者INT64常量,表示将输入的第1维度分为group组。
 - eps(double, 计算输入): DOUBLE常量,
out和rstdOut计算公式中的值。 - activateSilu(bool, 计算输入): BOOL常量,是否支持silu计算。如果设置为true,则表示groupnorm计算后继续silu计算。
 - out(aclTensor*, 计算输出): 输出张量,数据类型和shape与相同,支持ND,支持。
- [object Object]Atlas 推理系列产品[object Object]:数据类型支持FLOAT16、FLOAT。
 - [object Object]Atlas A2 训练系列产品/Atlas A2 推理系列产品[object Object]、[object Object]Atlas A3 训练系列产品/Atlas A3 推理系列产品[object Object]:数据类型支持FLOAT16、FLOAT、BFLOAT16。
 
 - meanOut(aclTensor*, 计算输出): 输出张量,shape为
(N, group),其中N与的第0维度保持一致,支持ND,支持。- [object Object]Atlas 推理系列产品[object Object]:数据类型与self相同。
 - [object Object]Atlas A2 训练系列产品/Atlas A2 推理系列产品[object Object]、[object Object]Atlas A3 训练系列产品/Atlas A3 推理系列产品[object Object]:数据类型与self相同。
 
 - rstdOut(aclTensor*, 计算输出): 输出张量,shape为
(N, group),其中N与的第0维度保持一致,支持ND,支持。- [object Object]Atlas 推理系列产品[object Object]:数据类型与self相同。
 - [object Object]Atlas A2 训练系列产品/Atlas A2 推理系列产品[object Object]、[object Object]Atlas A3 训练系列产品/Atlas A3 推理系列产品[object Object]:数据类型与self相同。
 
 - workspaceSize(uint64_t*, 出参): 返回需要在Device侧申请的workspace大小。
 - executor(aclOpExecutor **, 出参): 返回op执行器,包含算子计算流程。
 
- self(aclTensor*, 计算输入):
 返回值:
[object Object]
aclnnGroupNormSiluV2
参数说明:
- workspace(void*, 入参):在Device侧申请的workspace内存地址。
 - workspaceSize(uint64_t, 入参):在Device侧申请的workspace大小,由第一段接口aclnnGroupNormSiluV2GetWorkspaceSize获取。
 - executor(aclOpExecutor*, 入参):op执行器,包含了算子计算流程。
 - stream(aclrtStream, 入参):指定执行任务的Stream。
 
返回值:
约束说明
无
调用示例
[object Object]