Searched defs:cpu_client (Results 1 – 1 of 1) sorted by relevance
228 StatusOr<PjRtDevice*> DeviceForDLContext(const PjRtClient* cpu_client, in DeviceForDLContext()326 const pybind11::capsule& tensor, std::shared_ptr<PyClient> cpu_client, in DLPackManagedTensorToBuffer()