npu.set_npu_loop_size

Product	Supported
Atlas 350 Accelerator Card	√
Atlas A3 training product/Atlas A3 inference product	√
Atlas A2 training product/Atlas A2 inference product	√
Atlas 200I/500 A2 inference product	☓
Atlas inference product	☓
Atlas training product	√

Sets the number of iterations (or steps) per loop offloaded to the NPU.

npu.set_npu_loop_size(loop_size)

Parameter	Input/Output	Description
loop_size	Input	Number of steps per loop. Must be a positive integer.

None

import npu_device as npu
npu.set_npu_loop_size(100) # Set 100 steps to offload.

Parent topic: TF Adapter 2.x API Reference