aclnnUpsampleTrilinear3d
支持的产品型号
- Atlas 训练系列产品。
- Atlas A2训练系列产品/Atlas 800I A2推理产品。
接口原型
每个算子分为，必须先调用“aclnnUpsampleTrilinear3dGetWorkspaceSize”接口获取计算所需workspace大小以及包含了算子计算流程的执行器，再调用“aclnnUpsampleTrilinear3d”接口执行计算。
aclnnStatus aclnnUpsampleTrilinear3dGetWorkspaceSize(const aclTensor *self, const aclIntArray *outputSize, bool alignCorners, double scalesD, double scalesH, double scalesW, aclTensor *out, uint64_t *workspaceSize, aclOpExecutor **executor)aclnnStatus aclnnUpsampleTrilinear3d(void *workspace, uint64_t workspaceSize, aclOpExecutor *executor, aclrtStream stream)
功能描述
算子功能：对由多个输入通道组成的输入信号应用三线性插值算法进行上采样。
aclnnUpsampleTrilinear3dGetWorkspaceSize
参数说明：
self（const aclTensor*, 计算输入）：Device侧的aclTensor，数据类型支持FLOAT、FLOAT16、DOUBLE，shape仅支持五维，支持，支持NCDHW、NDHWC。
outputSize（const aclIntArray*, 计算输入）：INT64类型，指定输出Tensor大小，size大小为3，表示输入self在D、H和W维度上的空间大小。
alignCoreners（bool, 计算输入）: 是否对齐角像素点。如果为true，则输入和输出张量的角像素点会被对齐，否则不对齐。
scalesD（double, 计算输入）：double常量，表示输出out的depth维度乘数。
scalesH（double, 计算输入）：double常量，表示输出out的height维度乘数。
scalesW（double, 计算输入）：double常量，表示输出out的width维度乘数。
out（aclTensor*,计算输出）: 输出张量，数据类型支持FLOAT、FLOAT16、DOUBLE，shape仅支持五维，且数据类型与self的数据类型一致，支持，支持NCDHW、NDHWC。
workspaceSize（uint64_t*, 出参）：返回用户需要在Device侧申请的workspace大小。
executor（aclOpExecutor**, 出参）：返回op执行器，包含了算子计算流程。
返回值：
aclnnUpsampleTrilinear3d
参数说明：
- workspace（void*, 入参）：在Device侧申请的workspace内存地址。
- workspaceSize（uint64_t, 入参）：在Device侧申请的workspace大小，由第一段接口aclnnUpsampleTrilinear3dGetWorkspaceSize获取。
- executor（aclOpExecutor*, 入参）：op执行器，包含了算子计算流程。
- stream（aclrtStream, 入参）：指定执行任务的 AscendCL Stream流。
返回值：
约束与限制
无