aclnnRoiAlignV2Backward

产品支持情况

产品	是否支持
[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的反向传播，RoiAlign是一种池化层，用于非均匀输入尺寸的特征图，并输出固定尺寸的特征图。

函数原型

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

• aclnnStatus aclnnRoiAlignV2BackwardGetWorkspaceSize(const aclTensor* gradOutput, const aclTensor* boxes, const aclIntArray* inputShape, int64_t pooledHeight, int64_t pooledWidth, float spatialScale, int64_t samplingRatio, bool aligned, aclTensor* gradInput, uint64_t* workspaceSize, aclOpExecutor** executor)
• aclnnStatus aclnnRoiAlignV2Backward(void* workspace, uint64_t workspaceSize, aclOpExecutor* executor, aclrtStream stream)

aclnnRoiAlignV2BackwardGetWorkspaceSize

• 参数说明：

  • gradOutput（aclTensor*，计算输入）：反向传播的输入。Device侧的aclTensor，数据类型支持FLOAT，必须与boxes、gradInput数据类型一致。支持undefined，undefined支持NCHW。维度为4维，shape为（K，C，pooledHeight，pooledWidth），表示反向传播的输入梯度张量一个batch内有K个元素，每个元素有C个尺寸为pooledHeight * pooledWidth的特征图，K需要与boxes第0维保持一致。

  • boxes（aclTensor*，计算输入）：感兴趣区域box坐标。Device侧的aclTensor，数据类型支持FLOAT，必须与gradOutput、gradInput数据类型一致。支持undefined，undefined支持ND。维度为2维，shape为（K，5），5代表box相关信息（image_id，x1，y1，x2，y2）。image_id取值范围[0, B)，向下取整到图像id，B为inputShape第一个值。

  • inputShape（aclIntArray*，计算输入）：正向输入的shape，用来指定反向传播的输出shape。Host侧的aclIntArray，支持的数据类型为INT32、INT64。size大小为4，值为（B, C, inputHeight, inputWidth），表示正向RoiAlign的输入张量一个batch内有B张图像，每个图像有C个尺寸为inputHeight * inputWidth的特征图。

  • pooledHeight（int64_t，计算输入）：正向RoiAlign池化后输出图像的高度。Host侧的输入参数。

  • pooledWidth（int64_t，计算输入）：正向RoiAlign池化后输出图像的宽度。Host侧的输入参数。

  • spatialScale（float，计算输入）：乘法空间尺度因子，将ROI坐标从其输入空间尺度转换为池化时使用的尺度，即输入特征图X相对于输入图像的空间尺度。Host侧的输入参数，需大于0。

  • samplingRatio（int64_t，计算输入）：RoiAlign中用于计算每个输出元素在H和W方向上的采样频率。Host侧的输入参数，需大于等于0。

  • aligned（bool，计算输入）：如果为false，则对齐undefined版本实现；如果为true，则box坐标像素偏移-0.5来使相邻像素索引更好对齐。Host侧的输入参数。

  • gradInput（aclTensor*，计算输出）：反向传播的输出。Device侧的aclTensor，数据类型支持FLOAT，必须与gradOutput、boxes数据类型一致。支持undefined，undefined支持NCHW。维度为4维，shape为（B, C, inputHeight, inputWidth）。

  • workspaceSize（uint64_t*，出参）：返回需要在Device侧申请的workspace大小。

  • executor（aclOpExecutor**，出参）：返回op执行器，包含了算子计算流程。

• 返回码：

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

aclnnRoiAlignV2Backward

• 参数说明：

  • workspace（void*, 入参）：在Device侧申请的workspace内存地址。

  • workspaceSize（uint64_t, 入参）：在Device侧申请的workspace大小，由第一段接口aclnnRoiAlignV2BackwardGetWorkspaceSize获取。

  • executor（aclOpExecutor*, 入参）：op执行器，包含了算子计算流程。

  • stream（aclrtStream, 入参）：指定执行任务的Stream。

• 返回码：

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

约束说明

无

调用示例

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