aclnnReflectionPad1d
支持的产品型号
- [object Object]Atlas 推理系列产品[object Object]。
- [object Object]Atlas 训练系列产品[object Object]。
- [object Object]Atlas A2 训练系列产品/Atlas 800I A2 推理产品/A200I A2 Box 异构组件[object Object]。
- [object Object]Atlas A3 训练系列产品/Atlas A3 推理系列产品[object Object]。
功能说明
算子功能:使用输入边界的反射填充输入tensor。
示例:
[object Object]
函数原型
每个算子分为,必须先调用“aclnnReflectionPad1dGetWorkspaceSize”接口获取计算所需workspace大小以及包含了算子计算流程的执行器,再调用“aclnnReflectionPad1d”接口执行计算。
aclnnStatus aclnnReflectionPad1dGetWorkspaceSize(const aclTensor *self, const aclIntArray *padding, aclTensor *out, uint64_t *workspaceSize, aclOpExecutor **executor)
aclnnStatus aclnnReflectionPad1d(void *workspace, uint64_t workspaceSize, aclOpExecutor *executor, const aclrtStream stream)
aclnnReflectionPad1dGetWorkspaceSize
参数说明:
self(aclTensor*,计算输入):Device侧的aclTensor。支持,支持ND,维度支持二维或三维,在最后一维做pad。
- [object Object]Atlas A2 训练系列产品/Atlas 800I A2 推理产品/A200I A2 Box 异构组件[object Object]、[object Object]Atlas A3 训练系列产品/Atlas A3 推理系列产品[object Object]:FLOAT16、FLOAT32、DOUBLE、INT8、INT16、INT32、INT64、UINT8、BOOL、BFLOAT16
- [object Object]Atlas 推理系列产品[object Object]、[object Object]Atlas 训练系列产品[object Object]:FLOAT16、FLOAT32、DOUBLE、INT8、INT16、INT32、INT64、UINT8、BOOL
padding(aclIntArray*,计算输入):Device侧的aclIntArray数组,数据类型为INT64,长度为2,两个数值依次代表左右两边需要填充的值,且均需小于self最后一维度的数值。
out(aclTensor*,计算输出):Device侧的aclTensor,数据类型、、维度与self一致,out最后一维度的数值等于self最后一维度的数值加padding前两个值。支持。
- [object Object]Atlas A2 训练系列产品/Atlas 800I A2 推理产品/A200I A2 Box 异构组件[object Object]、[object Object]Atlas A3 训练系列产品/Atlas A3 推理系列产品[object Object]:FLOAT16、FLOAT32、DOUBLE、INT8、INT16、INT32、INT64、UINT8、BOOL、BFLOAT16
- [object Object]Atlas 推理系列产品[object Object]、[object Object]Atlas 训练系列产品[object Object]:FLOAT16、FLOAT32、DOUBLE、INT8、INT16、INT32、INT64、UINT8、BOOL
workspaceSize(uint64_t *,出参):返回需要在Device侧申请的workspace大小。
executor(aclOpExecutor **,出参):返回op执行器,包含了算子计算流程。
返回值:
[object Object]
aclnnReflectionPad1d
参数说明:
workspace(void *,入参):在Device侧申请的workspace内存地址。
workspaceSize(uint64_t,入参):在Device侧申请的workspace大小,由第一段接口aclnnReflectionPad1dGetWorkspaceSize获取。
executor(aclOpExecutor *,入参):op执行器,包含了算子计算流程。
stream(aclrtStream,入参):指定执行任务的AscendCL Stream流。
返回值:
约束说明
- 如果计算量过大可能会导致算子执行超时(aicore error类型报错,errorStr为:timeout or trap error),场景为最后2轴合轴小于16,前面的轴合轴超大。