xla

[benchmarks] Run some models with smaller batch sizes.

#6542

Merged