Fix docs referring to `--print-ir-before/after-all`, which has changed to `--mlir-print...`.
Fix docs referring to `--print-ir-before/after-all`, which has changed to `--mlir-print...`.

PiperOrigin-RevId: 448890307

google/jax,google/jax,google/jax,google/jax