xla
add test_set_batch_size flag
#982
Merged

Loading