| /external/tensorflow/tensorflow/compiler/xla/pjrt/ |
| D | tpu_client.h | 35 int process_index, const std::array<int, 3>& coords, in PjRtTpuDevice() argument
38 std::move(device_kind), process_index), in PjRtTpuDevice()
48 core_.Id(), process_index, coords_[0], in PjRtTpuDevice()
52 id(), process_index, absl::StrJoin(coords_, ","), core_on_chip()); in PjRtTpuDevice()
74 int process_index);
|
| D | tpu_client.cc | 87 int process_index) in PjRtTpuClient() argument
89 process_index, in PjRtTpuClient()
233 int process_index = topology.IdForHost(core.host_coordinates()); in GetTpuDevices() local
241 core, std::move(local_device_state), process_index, coords_array, in GetTpuDevices()
302 int process_index = platform->GetTpuHostLocation().Id(); in GetTpuClient() local
305 client, std::move(devices), process_index)); in GetTpuClient()
|
| D | tfrt_cpu_pjrt_client.h | 63 return process_index() == client()->process_index(); in IsAddressable()
68 int process_index() const override { return 0; } in process_index() function
115 TfrtCpuClient(int process_index,
120 int process_index() const override { return process_index_; } in process_index() function
|
| D | pjrt_stream_executor_client.h | 64 std::string device_kind, int process_index = 0)
69 process_index_(process_index), in id_()
83 int process_index() const override { return process_index_; } in process_index() function
151 int process_index, std::unique_ptr<se::DeviceMemoryAllocator> allocator,
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
141 return process_index_args.process_index; in process_index()
308 int PjRtCApiDevice::process_index() const { in process_index() function in xla::PjRtCApiDevice
315 return args.process_index; in process_index()
|
| D | pjrt_c_api_client.h | 44 int process_index() const override;
107 int process_index() const override;
|
| D | tfrt_cpu_pjrt_client.cc | 175 int process_index, std::vector<std::unique_ptr<TfrtCpuDevice>> devices, in TfrtCpuClient() argument
177 : process_index_(process_index), in TfrtCpuClient()
362 if (device->process_index() != process_index()) { in Compile()
1327 CHECK_EQ(device->process_index(), client_->process_index()); in ExecuteHelper()
|
| D | pjrt_client.h | 115 virtual int process_index() const = 0;
438 virtual int process_index() const = 0;
|
| D | pjrt_stream_executor_client.cc | 208 int process_index, std::unique_ptr<se::DeviceMemoryAllocator> allocator, in PjRtStreamExecutorClient() argument
218 process_index_(process_index), in PjRtStreamExecutorClient()
2111 CHECK_EQ(device->process_index(), client_->process_index()); in ExecuteHelper()
2422 if (device->process_index() != process_index()) { in GetExecutableExtras()
|
| D | gpu_device.cc | 415 process_index()); in GpuDevice()
|
| /external/tensorflow/tensorflow/compiler/xla/python/tpu_driver/client/ |
| D | tpu_client.cc | 41 TpuDevice::TpuDevice(int id, int process_index, in TpuDevice() argument
44 process_index_(process_index), in TpuDevice()
65 int process_index = chip.host_id(); in GetTpuDevices() local
68 core.id(), process_index, coords_array, core.core_on_chip_index()); in GetTpuDevices()
101 int process_index) in PyTpuClient() argument
106 process_index_(process_index) { in PyTpuClient()
112 if (device->process_index() == process_index_) { in PyTpuClient()
536 if (device->process_index() != client_->process_index()) { in PyTpuExecutable()
561 CHECK_EQ(device->process_index(), client_->process_index()); in ExecuteHelper()
|
| D | tpu_client_extension.cc | 44 .def("process_index", &PyTpuClient::process_index) in PYBIND11_MODULE()
45 .def("host_id", &PyTpuClient::process_index) in PYBIND11_MODULE()
46 .def("task_id", &PyTpuClient::process_index) in PYBIND11_MODULE()
282 device.id(), device.process_index(), device.coords()[0], in PYBIND11_MODULE()
|
| D | tpu_client.h | 52 TpuDevice(int id, int process_index, const std::array<int, 3>& coords,
73 int process_index() const override { return process_index_; } in process_index() function
120 int process_index);
143 int process_index() const { return process_index_; } in process_index() function
|
| /external/tensorflow/tensorflow/compiler/xla/python/ |
| D | xla.cc | 132 "process_index", &PjRtDevice::process_index, in PYBIND11_MODULE()
135 .def_property_readonly("host_id", &PjRtDevice::process_index, in PYBIND11_MODULE()
137 .def_property_readonly("task_id", &PjRtDevice::process_index, in PYBIND11_MODULE()
210 .def("process_index", &PyClient::process_index) in PYBIND11_MODULE()
211 .def("host_id", &PyClient::process_index) in PYBIND11_MODULE()
212 .def("task_id", &PyClient::process_index) in PYBIND11_MODULE()
|
| D | py_client.h | 113 int process_index() const { return pjrt_client_->process_index(); } in process_index() function
|
| /external/tensorflow/tensorflow/compiler/xla/pjrt/c/ |
| D | pjrt_c_api.h | 235 int process_index; // out member
238 PJRT_STRUCT_SIZE(PJRT_Client_ProcessIndex_Args, process_index);
388 int process_index; // out member
391 PJRT_STRUCT_SIZE(PJRT_Device_ProcessIndex_Args, process_index);
|
| D | pjrt_c_api_wrapper_impl.cc | 108 args->process_index = args->client->client->process_index(); in PJRT_Client_ProcessIndex()
268 args->process_index = args->device->device->process_index(); in PJRT_Device_ProcessIndex()
|
| /external/perfetto/src/trace_processor/ |
| D | export_json.cc | 1677 for (uint32_t process_index = 0; in ExportMemorySnapshots() local
1678 process_index < process_snapshots.row_count(); ++process_index) { in ExportMemorySnapshots()
1679 if (process_snapshots.snapshot_id()[process_index].value != snapshot_id) in ExportMemorySnapshots()
1682 auto process_snapshot_id = process_snapshots.id()[process_index].value; in ExportMemorySnapshots()
1683 uint32_t pid = UpidToPid(process_snapshots.upid()[process_index]); in ExportMemorySnapshots()
|
| /external/python/setuptools/setuptools/tests/ |
| D | test_packageindex.py | 106 index.process_index(url, page)
|
| /external/tensorflow/tensorflow/compiler/xla/python/xla_extension/ |
| D | __init__.pyi | 309 process_index: int
358 def process_index(self) -> int: ...
|
| /external/python/setuptools/setuptools/ |
| D | package_index.py | 357 page = self.process_index(url, page)
432 def process_index(self, url, page): member in PackageIndex
|