Home
last modified time | relevance | path

Searched defs:addressable_device_logical_ids (Results 1 – 6 of 6) sorted by relevance

/aosp_15_r20/external/tensorflow/tensorflow/compiler/xla/pjrt/
H A Dpjrt_stream_executor_client.h334 addressable_device_logical_ids; member
746 absl::Span<const LogicalDeviceIds> addressable_device_logical_ids() in addressable_device_logical_ids() function
H A Dpjrt_stream_executor_client.cc1765 std::vector<LogicalDeviceIds> addressable_device_logical_ids, in PjRtStreamExecutorExecutable()
2394 addressable_device_logical_ids = extras.addressable_device_logical_ids; in GetExecutableExtras() local
2457 addressable_device_logical_ids = extras.addressable_device_logical_ids; in Compile() local
H A Dpjrt_c_api_client.h450 absl::Span<const LogicalDeviceIds> addressable_device_logical_ids() in addressable_device_logical_ids() function
H A Dtfrt_cpu_pjrt_client.h500 absl::Span<const LogicalDeviceIds> addressable_device_logical_ids() in addressable_device_logical_ids() function
H A Dtfrt_cpu_pjrt_client.cc353 addressable_device_logical_ids; in Compile() local
1164 std::vector<LogicalDeviceIds> addressable_device_logical_ids, in TfrtCpuExecutable()
/aosp_15_r20/external/tensorflow/tensorflow/compiler/xla/python/
H A Dpy_executable.h86 addressable_device_logical_ids() const { in addressable_device_logical_ids() function