tree 0aed1643b5e88a97ff5f2213ef9ad25bfa835fef
parent 0f2ca64a4fa6bc78d0819dc7ea3272882024d0e0
author A. Unique TensorFlower <gardener@tensorflow.org> 1658351595 -0700
committer TensorFlower Gardener <gardener@tensorflow.org> 1658351866 -0700

[PJRT:C] Add PjRtExecutable::Execute to PJRT C API.

- struct PJRT_ExecuteOptions is added with minimum fields (launch_id).
- The input argument_handles is a 2D vector of PjRtCApiBuffer* (size [num_devices, num_args]. It is converted to PJRT_Buffer*** in pjrt_c_api_client.cc, and later converted from PJRT_Buffer*** to 2D vector of PjRtBuffer* in pjrt_c_api_wrapper_impl.cc.
- The output buffers is a 2D vector of PjRtBuffer* (size [num_devices, num_outputs]). pjrt_c_api_client.cc allocates 2D vectors of PJRT_Buffer* in args.output_lists. pjrt_c_api_wrapper_impl.cc converts 2D vector of PjRtBuffer* to PJRT_Buffer* and populates args.output_lists. pjrt_c_api_client.cc converts PJRT_Buffer*** to 2D vector of PjRtBuffer* and deallocate PJRT_Buffer*** and PJRT_Buffer**.

PiperOrigin-RevId: 462227365