Allow `steps_per_loop` in Controller to be passed as a callable.
PiperOrigin-RevId: 466412169
R
Ruoxin Sang committed
90dedf267cacda37581ab51fb674a00efe289696
Parent: db19ab9
Committed by A. Unique TensorFlower <gardener@tensorflow.org>
on 8/9/2022, 6:12:56 PM