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

RefExit

产品支持情况

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

功能说明

退出当前帧(frame)到其父帧。RefExit算子与Enter算子配合使用在图中创建循环,将可被引用覆盖的输入张量x传递到父帧使其可用。与Exit算子不同,RefExit作用于可被ref类算子覆盖的引用张量。输出张量y与输入张量x具有相同的形状和数据类型。

参数说明

参数名 输入/输出/属性 描述 数据类型 数据格式
x 输入 输入张量。需要传递到父帧的可被引用覆盖的张量。 float16、float32、float64、int8、int16、int32、int64、uint8、uint16、uint32、uint64、bool ND
y 输出 与输入张量x具有相同形状和数据类型的张量。 float16、float32、float64、int8、int16、int32、int64、uint8、uint16、uint32、uint64、bool ND

约束说明

  • 输出张量y的数据类型和形状必须与输入张量x一致。

调用说明

RefExit算子为控制流算子,通常与Enter算子配合使用在图中创建循环,不单独调用。

调用方式 调用样例 说明
图模式调用 test_geir_ref_exit 通过算子IR构图方式调用RefExit算子