xla

afaab1f6 - [PJRT] Enable PJRT C API option (#4077)

[PJRT] Enable PJRT C API option (#4077) * Enable PJRT C API option * Add patch for `pjrt_c_api_client.cc` * Remove obsolete comment. * Only use `on_device_shape()` with C API