Searched defs:process_index (Results 1 – 11 of 11) sorted by relevance
87 int process_index) in PjRtTpuClient()233 int process_index = topology.IdForHost(core.host_coordinates()); in GetTpuDevices() local302 int process_index = platform->GetTpuHostLocation().Id(); in GetTpuClient() local
35 int process_index, const std::array<int, 3>& coords, in PjRtTpuDevice() argument
68 int process_index() const override { return 0; } in process_index() function120 int process_index() const override { return process_index_; } in process_index() function
83 int process_index() const override { return process_index_; } in process_index() function157 int process_index() const override { return process_index_; } in process_index() function
133 int PjRtCApiClient::process_index() const { in process_index() function in xla::PjRtCApiClient308 int PjRtCApiDevice::process_index() const { in process_index() function in xla::PjRtCApiDevice
175 int process_index, std::vector<std::unique_ptr<TfrtCpuDevice>> devices, in TfrtCpuClient()
208 int process_index, std::unique_ptr<se::DeviceMemoryAllocator> allocator, in PjRtStreamExecutorClient()
113 int process_index() const { return pjrt_client_->process_index(); } in process_index() function
73 int process_index() const override { return process_index_; } in process_index() function143 int process_index() const { return process_index_; } in process_index() function
41 TpuDevice::TpuDevice(int id, int process_index, in TpuDevice()65 int process_index = chip.host_id(); in GetTpuDevices() local101 int process_index) in PyTpuClient()
235 int process_index; // out member388 int process_index; // out member