Searched defs:local_device_count (Results 1 – 4 of 4) sorted by relevance
72 def local_device_count(self): member in TpuBackend
83 int local_device_count() const { return local_devices_.size(); } in local_device_count() function
136 int local_device_count() const { return local_devices_.size(); } in local_device_count() function
62 def local_device_count(self): member in Backend120 def local_device_count(self): member in LocalBackend