FilesLast commitLast update
13 days ago
2 months ago
3 months ago
12 days ago
17 days ago
12 days ago
12 days ago
29 days ago
29 days ago
12 days ago
12 days ago
README

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算子