昇腾社区首页
中文
注册
开发者
下载

TensorScatterUpdateFusionPass

融合模式

该融合将符合图融合pattern的TensorScatterUpdate算子,在输入输出数据类型不是Bool或int32时,修改为TensorMove+ScatterNdUpdate算子。

使用约束

  • X的输入类型是Bool或者int32类型时,该融合规则不生效。
  • 该融合规则不可关闭。
  • 由于ScatterNdUpdate算子aicore实现支持的数据类型为:int64、int8、float32、float16、bfloat16和bool。因此融合成功后,只有这六种数据类型执行aicore,其他类型执行aicpu。
  • String和complex128类型在tensormove/ScatterNdUpdate上不支持时,该融合规则不生效。

支持的型号

Atlas A2 训练系列产品/Atlas A2 推理系列产品