CANN ops-math AdaCast算子

AdaCast

【免费下载链接】ops-math本项目是CANN提供的数学类基础计算算子库，实现网络在NPU上加速计算。项目地址: https://gitcode.com/cann/ops-math

产品支持情况

功能说明

• 算子功能：面向 HDRnet（High Dynamic Range Network）实时图像增强网络的输入预处理算子。将传感器输出的 uint16 HDR 图像数据转换为 float16 张量，并按白电平（pixel）完成归一化，等价于把「整型→浮点类型转换」与「按除数缩放」两步融合为单次逐元素运算。
• 计算公式：

$$ out_{i} = \mathrm{Cast_FP16}^{sat+round}\bigl(\mathrm{Cast_FP32}(self_{i}) \times (1 / pixel)\bigr) $$

其中末步采用饱和（clip 到 fp16 值域 [-65504, 65504]）+ 四舍五入（round-to-nearest-even）方式转换到 float16。

参数说明

约束说明

• 数据格式仅支持 ND。
• 输入 x 的 rank 范围为 [1, 4]，输出 y 的 shape 必须与 x 完全一致（element-wise 语义，不支持广播）。
• 类型组合固定为 UINT16 → FLOAT16，不支持其他 dtype 组合。
• 属性 pixel 必须为正整数（pixel > 0），否则返回 ACLNN_ERR_PARAM_INVALID。
• 支持空 Tensor（0 元素），直接返回。
• 默认确定性实现，相同输入始终产生相同输出。

调用说明

【免费下载链接】ops-math本项目是CANN提供的数学类基础计算算子库，实现网络在NPU上加速计算。项目地址: https://gitcode.com/cann/ops-math