Searched refs:addressable_devices_ (Results 1 – 2 of 2) sorted by relevance
137 return addressable_devices_.size(); in addressable_device_count()141 return addressable_devices_; in addressable_devices()216 addressable_devices_.at(device_ordinal)) in device_state()262 std::vector<PjRtDevice*> addressable_devices_; variable643 return addressable_devices_; in addressable_devices()731 std::vector<PjRtDevice*> addressable_devices_; variable
241 if (idx >= addressable_devices_.size()) { in PjRtStreamExecutorClient()242 addressable_devices_.resize(idx + 1); in PjRtStreamExecutorClient()244 CHECK(addressable_devices_[idx] == nullptr) << idx; in PjRtStreamExecutorClient()245 addressable_devices_[idx] = device.get(); in PjRtStreamExecutorClient()249 for (int idx = 0; idx < addressable_devices_.size(); ++idx) { in PjRtStreamExecutorClient()250 CHECK(addressable_devices_[idx] != nullptr) << idx; in PjRtStreamExecutorClient()988 for (auto* device : addressable_devices_) { in LookupAddressableDevice()1578 addressable_devices_(std::move(addressable_devices)) { in PjRtStreamExecutorExecutable()1589 CHECK(addressable_devices_.empty()); in PjRtStreamExecutorExecutable()1594 CHECK_GE(addressable_devices_.size(), 1) << device_assignment_->ToString(); in PjRtStreamExecutorExecutable()[all …]