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