Add batches argument to throughput (#18905)

Commit

2 years ago

Add batches argument to throughput (#18905) Co-authored-by: Adrian Wälchli <aedu.waelchli@gmail.com>