Searched refs:DeviceForDLContext (Results 1 – 1 of 1) sorted by relevance
225 StatusOr<PjRtDevice*> DeviceForDLContext(const PjRtClient& client, in DeviceForDLContext() function334 DeviceForDLContext(*client->pjrt_client(), dlmt->dl_tensor.ctx)); in DLPackManagedTensorToBuffer()