Add
产品支持情况
| 产品 | 是否支持 |
|---|---|
| Ascend 950PR&950DT系列产品 | √ |
| Atlas A3系列产品 | √ |
| Atlas A2系列产品 | √ |
| Atlas 200I/500 A2推理产品 | × |
| Atlas推理系列产品 | × |
| Atlas训练系列产品 | √ |
| Kirin X90处理器系列产品 | √ |
| Kirin 9030处理器系列产品 | √ |
功能说明
- 算子功能:对输入完成相加操作。
- 计算公式:
y=x1+x2y = x_1 + x_2
参数说明
| 参数名 | 输入/输出/属性 | 描述 | 数据类型 | 数据格式 |
|---|---|---|---|---|
| x1 | 输入 | 公式中的输入张量x_1 | BOOL, INT8, INT16, INT32, INT64, UINT8, FLOAT64, FLOAT16, BFLOAT16, FLOAT32, COMPLEX128, COMPLEX64 | ND |
| x2 | 输入 | 公式中的输入张量x2 | 同x1 | ND |
| y | 输出 | 公式中的输出张量y | 同x1 | ND |
约束说明
- 不支持 STRING 数据类型。算子IR(add_proto.h)中注册 DT_STRING 仅为兼容 TensorFlow Add 算子,图模式(GEIR)调用通路虽可通过类型推断,但无对应kernel实现,运行时会失败。
- COMPLEX32 仅 Ascend 950 内核实现支持,aclnnAdd / aclnnInplaceAdd 接口及图模式通路均不支持,请使用 COMPLEX64 / COMPLEX128。
- Kirin X90/Kirin 9030系列产品支持INT8,UINT8,INT32,FLOAT16,FLOAT32数据类型
调用说明
| 调用方式 | 调用样例 | 说明 |
|---|---|---|
| aclnn调用 | test_aclnn_add | 通过aclnnAdd和aclnnInplaceAdd接口方式调用Add算子 |
| 图模式调用 | test_geir_add | 通过算子IR构图方式调用Add算子 |