Home
last modified time | relevance | path

Searched defs:LookupAddressableDevice (Results 1 – 3 of 3) sorted by relevance

/aosp_15_r20/external/tensorflow/tensorflow/compiler/xla/pjrt/
H A Dpjrt_c_api_client.h117 StatusOr<PjRtDevice*> LookupAddressableDevice( in LookupAddressableDevice() function
H A Dtfrt_cpu_pjrt_client.cc221 StatusOr<PjRtDevice*> TfrtCpuClient::LookupAddressableDevice( in LookupAddressableDevice() function in xla::TfrtCpuClient
H A Dpjrt_stream_executor_client.cc1103 StatusOr<PjRtDevice*> PjRtStreamExecutorClient::LookupAddressableDevice( in LookupAddressableDevice() function in xla::PjRtStreamExecutorClient