Searched defs:local_device_states (Results 1 – 2 of 2) sorted by relevance
138 std::vector<std::unique_ptr<LocalDeviceState>> local_device_states) { in GetTpuDevices()206 std::vector<std::unique_ptr<LocalDeviceState>> local_device_states; in GetTpuClient() local
230 std::vector<std::unique_ptr<LocalDeviceState>> local_device_states) { in BuildLocalDevices()245 std::vector<std::unique_ptr<LocalDeviceState>> local_device_states, in BuildDistributedDevices()