FilesLast commitLast update
15 days ago
15 days ago
15 days ago
15 days ago
15 days ago
15 days ago
README

For

产品支持情况

产品 是否支持
Ascend 950PR&950DT系列产品 √
Atlas A3系列产品 √
Atlas A2系列产品 √
Atlas 200I/500 A2推理产品 √
Atlas推理系列产品 √
Atlas训练系列产品 √

功能说明

  • 算子功能:循环执行body子图,循环变量从start开始,每次迭代递增delta,直到循环变量超过limit。每次迭代时,将input传入body子图,body返回的输出作为下一次迭代的输入,最终输出为最后一次迭代的结果。

参数说明

参数名 输入/输出/属性 描述 数据类型 数据格式
start 输入 循环下界,int32标量。 INT32 ND
limit 输入 循环上界,int32标量。 INT32 ND
delta 输入 步长大小,int32标量。 INT32 ND
input 输入(动态) 输入张量,支持动态输入。 所有类型 ND
body 子图 接收input并返回另一组张量的子图,用于计算下一次迭代的值。 - -
output 输出(动态) body子图返回的输出张量,类型与input相同,支持动态输出。 所有类型 ND

约束说明

  • start、limit、delta必须为INT32标量。
  • input的数量必须与output的数量一致。
  • 各input与对应output的数据类型需保持一致。

调用说明

调用方式 调用样例 说明
图模式调用 test_geir_for 通过算子IR构图方式调用For算子。