Home
last modified time | relevance | path

Searched defs:LookupDevice (Results 1 – 8 of 8) sorted by relevance

/aosp_15_r20/external/intel-media-driver/media_softlet/agnostic/common/media_interfaces/
H A Dskuwa_factory.h78 static Type LookupDevice(KeyType key) in LookupDevice() function
/aosp_15_r20/external/tensorflow/tensorflow/core/common_runtime/
H A Ddevice_mgr.cc112 Status StaticDeviceMgr::LookupDevice(StringPiece name, Device** device) const { in LookupDevice() function in tensorflow::StaticDeviceMgr
H A Ddynamic_device_mgr.cc98 Status DynamicDeviceMgr::LookupDevice(StringPiece name, Device** device) const { in LookupDevice() function in tensorflow::DynamicDeviceMgr
H A Dgraph_execution_state.cc352 Status LookupDevice(const DeviceSet& device_set, const string& tensor_name, in LookupDevice() function
/aosp_15_r20/external/tensorflow/tensorflow/compiler/xla/pjrt/
H A Dpjrt_stream_executor_client.h168 StatusOr<PjRtDevice*> LookupDevice(int device_id) const override { in LookupDevice() function
H A Dpjrt_c_api_client.cc161 StatusOr<PjRtDevice*> PjRtCApiClient::LookupDevice(int device_id) const { in LookupDevice() function in xla::PjRtCApiClient
H A Dtfrt_cpu_pjrt_client.cc212 StatusOr<PjRtDevice*> TfrtCpuClient::LookupDevice(int device_id) const { in LookupDevice() function in xla::TfrtCpuClient
/aosp_15_r20/external/tensorflow/tensorflow/compiler/xla/python/tpu_driver/client/
H A Dtpu_client.cc512 static std::shared_ptr<PjRtDevice> LookupDevice(const PyTpuClient& client, in LookupDevice() function