特性范围
数据类型
FWD_D、BWD_D支持f32/f16/bf16数据类型的任意组合。
传播方向 |
src数据类型 |
dst数据类型 |
diff_dst数据类型 |
diff_src数据类型 |
|---|---|---|---|---|
FWD_D、BWD_D |
f32 |
f32 |
f32 |
f32 |
f16 |
f16 |
f16 |
f16 |
|
bf16 |
bf16 |
bf16 |
bf16 |
数据排布
支持3D~5D Tensor，支持以下数据排布，输入和输出的排布相同。
Tensor维度 |
tag(src/dst/diff_dst/diff_src) |
|---|---|
3D Tensor |
abc,acb |
4D Tensor |
abcd,acdb |
5D Tensor |
abcde,acdeb |
参数约束
字段 |
取值 |
|---|---|
--dir |
FWD_D [default], BWD_D |
--sdt |
f32 [default], f16, bf16, s32, s8, u8 |
--ddt |
f32 [default], f16, bf16, s32, s8, u8 |
--alg |
nearest [default], linear |
--tag |
axb [default], abx |
resampling要求输入输出内存排布相同，但是维度大小可以不同。
具体参考如下用例：
- 5D维度：mb4_ic8_id4od8_ih4oh8_iw4ow8，输入4x8x4x4x4，输出4x8x8x8x8
- 4D维度：mb4_ic8_ih4oh8_iw4ow8，输入4x8x4x4，输出4x8x8x8
- 3D维度：mb4_ic8_iw4ow8，输入4x8x4，输出4x8x8
另外id/ih/iw/od/oh/ow的某一项过大可能会出现精度错误，由于测试系统限制，其他平台表现相同，不属于功能性问题；如需规避上述问题，建议将上述提到的参数规模控制在20000以内。
父主题： Resampling