Searched defs:local_device_states (Results 1 – 2 of 2) sorted by relevance
323 std::map<int, std::unique_ptr<LocalDeviceState>> local_device_states) { in BuildLocalDevices()338 std::map<int, std::unique_ptr<LocalDeviceState>> local_device_states, in BuildDistributedDevices()429 std::map<int, std::unique_ptr<LocalDeviceState>> local_device_states; in GetGpuClient() local
213 std::vector<std::unique_ptr<LocalDeviceState>> local_device_states) { in GetTpuDevices()291 std::vector<std::unique_ptr<LocalDeviceState>> local_device_states; in GetTpuClient() local