xla

3dfd0f9b - Add a patch for mismatched batch size. (#6029)