NFC: Split launch grid and validation out of MatMul emitter.

Also
- Unify naming to ${scope}_${thing}_dim_idx.
- Make logic for split_lhs_nc a bit easier to follow (at least in my opinion), by
  decoupling it from the batch_size computation.

PiperOrigin-RevId: 564435916