Searched defs:process_index (Results 1 – 13 of 13) sorted by relevance
| /external/tensorflow/tensorflow/compiler/xla/pjrt/ |
| D | tpu_client.cc | 87 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
|
| D | tpu_client.h | 35 int process_index, const std::array<int, 3>& coords, in PjRtTpuDevice() argument
|
| D | tfrt_cpu_pjrt_client.h | 68 int process_index() const override { return 0; } in process_index() function
120 int process_index() const override { return process_index_; } in process_index() function
|
| D | pjrt_stream_executor_client.h | 83 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
|
| D | pjrt_c_api_client.cc | 133 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
|
| D | tfrt_cpu_pjrt_client.cc | 175 int process_index, std::vector<std::unique_ptr<TfrtCpuDevice>> devices, in TfrtCpuClient()
|
| D | pjrt_stream_executor_client.cc | 208 int process_index, std::unique_ptr<se::DeviceMemoryAllocator> allocator, in PjRtStreamExecutorClient()
|
| /external/tensorflow/tensorflow/compiler/xla/python/ |
| D | py_client.h | 113 int process_index() const { return pjrt_client_->process_index(); } in process_index() function
|
| /external/tensorflow/tensorflow/compiler/xla/python/tpu_driver/client/ |
| D | tpu_client.h | 73 int process_index() const override { return process_index_; } in process_index() function
143 int process_index() const { return process_index_; } in process_index() function
|
| D | tpu_client.cc | 41 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()
|
| /external/python/setuptools/setuptools/ |
| D | package_index.py | 432 def process_index(self, url, page): member in PackageIndex
|
| /external/tensorflow/tensorflow/compiler/xla/pjrt/c/ |
| D | pjrt_c_api.h | 235 int process_index; // out member
388 int process_index; // out member
|
| /external/perfetto/src/trace_processor/ |
| D | export_json.cc | 1677 for (uint32_t process_index = 0; in ExportMemorySnapshots() local
|