Home
last modified time | relevance | path

Searched refs:LookupAddressableDevice (Results 1 – 4 of 4) sorted by relevance

/external/tensorflow/tensorflow/compiler/xla/python/
Ddlpack.cc234 return client.LookupAddressableDevice(context.device_id); in DeviceForDLContext()
241 return client.LookupAddressableDevice(context.device_id); in DeviceForDLContext()
/external/tensorflow/tensorflow/compiler/xla/pjrt/
Dpjrt_client.h167 virtual StatusOr<PjRtDevice*> LookupAddressableDevice(
Dpjrt_stream_executor_client.h153 StatusOr<PjRtDevice*> LookupAddressableDevice(
Dpjrt_stream_executor_client.cc986 StatusOr<PjRtDevice*> PjRtStreamExecutorClient::LookupAddressableDevice( in LookupAddressableDevice() function in xla::PjRtStreamExecutorClient