Home
last modified time | relevance | path

Searched defs:addressable_device_count (Results 1 – 4 of 4) sorted by relevance

/aosp_15_r20/external/tensorflow/tensorflow/compiler/xla/python/
H A Dpy_client.h109 int addressable_device_count() const { in addressable_device_count() function
/aosp_15_r20/external/tensorflow/tensorflow/compiler/xla/pjrt/
H A Dtfrt_cpu_pjrt_client.h124 int addressable_device_count() const override { in addressable_device_count() function
H A Dpjrt_stream_executor_client.h160 int addressable_device_count() const override { in addressable_device_count() function
H A Dpjrt_c_api_client.cc110 int PjRtCApiClient::addressable_device_count() const { in addressable_device_count() function in xla::PjRtCApiClient