Searched refs:addressable_device_count (Results 1 – 5 of 5) sorted by relevance
100 int addressable_device_count() const { in addressable_device_count() function101 return pjrt_client_->addressable_device_count(); in addressable_device_count()
206 .def("local_device_count", &PyClient::addressable_device_count) in PYBIND11_MODULE()
153 virtual int addressable_device_count() const = 0;
136 int addressable_device_count() const override { in addressable_device_count() function
1595 CHECK_LE(addressable_devices_.size(), client_->addressable_device_count()) in PjRtStreamExecutorExecutable()