文件最后提交记录最后更新时间
1 个月前
1 个月前
4 个月前
1 个月前
1 个月前
1 个月前
3 个月前
1 个月前
README

DeformableOffsetsGrad

产品支持情况

产品 是否支持
Ascend 950PR/Ascend 950DT
Atlas A3 训练系列产品/Atlas A3 推理系列产品
Atlas A2 训练系列产品/Atlas A2 推理系列产品

功能说明

  • 算子功能:DeformableOffsets算子的目的是根据offsets(kernel采样点的偏移值)来收集用于卷积的特征采样点,并对其进行重组,方便Conv2d算子进行卷积计算。而DeformableOffsetsGrad即为这一过程的反向。

参数说明

参数名 输入/输出/属性 描述 数据类型 数据格式
grad 输入 输入的梯度。 DT_FLOAT16, DT_FLOAT, DT_BF16 NHWC
x 输入 输入的坐标。 DT_FLOAT16, DT_FLOAT, DT_BF16 NHWC
offsets 输入 偏移值。 DT_FLOAT16, DT_FLOAT, DT_BF16 NHWC
grad_x 输出 x的梯度。 DT_FLOAT16, DT_FLOAT NHWC
grad_offsets 输出 相对于input_offsets的梯度。 DT_FLOAT16, DT_FLOAT NHWC
strides 属性 滑动窗口的步长,包含4个整数的列表。 INT64 -
pads 属性 填充维度,包含4个整数的列表。 INT64 -
ksize 属性 卷积核的大小,包含2个整数列表。 INT64 -
dilations 属性 每个维度的伸缩因子,包含4个整数的列表。 INT64 -
data_format 属性 输入x的数据格式。 STRING -
deformable_groups 属性 C轴上的分组。 INT32 -
modulated 属性 可选bool指定DeformableConv2D的版本,true表示v2 ,false表示v1,只支持" true "。 BOOL -

约束说明

调用说明

调用方式 样例代码 说明
图模式 test_geir_deformable_offsets_grad 通过算子IR构图方式调用DeformableOffsetsGrad算子。