Home
last modified time | relevance | path

Searched defs:process_index (Results 1 – 16 of 16) sorted by relevance

/aosp_15_r20/external/tensorflow/tensorflow/compiler/xla/pjrt/
H A Dtpu_client.cc87 int process_index) in PjRtTpuClient()
233 int process_index = topology.IdForHost(core.host_coordinates()); in GetTpuDevices() local
302 int process_index = platform->GetTpuHostLocation().Id(); in GetTpuClient() local
H A Dtfrt_cpu_pjrt_client.h68 int process_index() const override { return 0; } in process_index() function
120 int process_index() const override { return process_index_; } in process_index() function
H A Dpjrt_stream_executor_client.h83 int process_index() const override { return process_index_; } in process_index() function
157 int process_index() const override { return process_index_; } in process_index() function
H A Dpjrt_c_api_client.cc133 int PjRtCApiClient::process_index() const { in process_index() function in xla::PjRtCApiClient
308 int PjRtCApiDevice::process_index() const { in process_index() function in xla::PjRtCApiDevice
H A Dtpu_client.h35 int process_index, const std::array<int, 3>& coords, in PjRtTpuDevice() argument
H A Dtfrt_cpu_pjrt_client.cc175 int process_index, std::vector<std::unique_ptr<TfrtCpuDevice>> devices, in TfrtCpuClient()
H A Dpjrt_stream_executor_client.cc208 int process_index, std::unique_ptr<se::DeviceMemoryAllocator> allocator, in PjRtStreamExecutorClient()
/aosp_15_r20/external/tensorflow/tensorflow/compiler/xla/python/tpu_driver/client/
H A Dtpu_client.cc41 TpuDevice::TpuDevice(int id, int process_index, in TpuDevice()
65 int process_index = chip.host_id(); in GetTpuDevices() local
101 int process_index) in PyTpuClient()
H A Dtpu_client.h73 int process_index() const override { return process_index_; } in process_index() function
143 int process_index() const { return process_index_; } in process_index() function
/aosp_15_r20/external/tensorflow/tensorflow/compiler/xla/python/
H A Dpy_client.h113 int process_index() const { return pjrt_client_->process_index(); } in process_index() function
/aosp_15_r20/external/tensorflow/tensorflow/compiler/xla/pjrt/c/
H A Dpjrt_c_api.h235 int process_index; // out member
388 int process_index; // out member
/aosp_15_r20/prebuilts/clang/host/linux-x86/clang-r530567b/python3/lib/python3.11/site-packages/setuptools/
Dpackage_index.py432 def process_index(self, url, page): member in PackageIndex
/aosp_15_r20/prebuilts/clang/host/linux-x86/clang-r522817/python3/lib/python3.11/site-packages/setuptools/
Dpackage_index.py432 def process_index(self, url, page): member in PackageIndex
/aosp_15_r20/prebuilts/clang/host/linux-x86/clang-r530567/python3/lib/python3.11/site-packages/setuptools/
Dpackage_index.py432 def process_index(self, url, page): member in PackageIndex
/aosp_15_r20/prebuilts/clang/host/linux-x86/clang-r536225/python3/lib/python3.11/site-packages/setuptools/
Dpackage_index.py432 def process_index(self, url, page): member in PackageIndex
/aosp_15_r20/external/python/setuptools/setuptools/
Dpackage_index.py432 def process_index(self, url, page): member in PackageIndex