Home
last modified time | relevance | path

Searched refs:c_api_device (Results 1 – 2 of 2) sorted by relevance

/aosp_15_r20/external/tensorflow/tensorflow/compiler/xla/pjrt/
H A Dpjrt_c_api_client.h84 static PjRtDevice* GetWrapped(PjRtDevice* c_api_device) { in GetWrapped() argument
85 return tensorflow::down_cast<PjRtCApiDevice*>(c_api_device)->wrapped(); in GetWrapped()
H A Dpjrt_c_api_client.cc445 PjRtCApiDevice* c_api_device = client_->GetCppDevice(device); in InitDevices() local
446 addressable_devices_.push_back(c_api_device); in InitDevices()