npu.set_npu_loop_size
函数原型
npu.set_npu_loop_size(loop_size)
功能说明
用于设置NPU循环下沉执行时的循环次数。
参数说明
参数名 |
输入/输出 |
描述 |
|---|---|---|
loop_size |
输入 |
NPU循环下沉执行时的循环次数，必须为大于0的整数。 |
返回值
无。
调用示例
import npu_device as npu npu.set_npu_loop_size(100) # 设置循环下沉次数为100
父主题： TF Adapter 2.6 接口参考
