Searched refs:gpu_devices (Results 1 – 2 of 2) sorted by relevance
62 std::map<int, string> gpu_devices; // GPU device map: id -> device name. in VirtualPlacer() local73 gpu_devices[parsed_name.id] = cluster_device_name; in VirtualPlacer()80 if (!gpu_devices.empty()) { in VirtualPlacer()82 default_device_name_ = gpu_devices.begin()->second; in VirtualPlacer()
54 gpu_devices = [57 self.devices = ['/cpu:0'] + gpu_devices