Searched refs:gpu_devices (Results 1 – 3 of 3) sorted by relevance
60 std::map<int, string> gpu_devices; // GPU device map: id -> device name. in VirtualPlacer() local71 gpu_devices[parsed_name.id] = cluster_device_name; in VirtualPlacer()78 if (!gpu_devices.empty()) { in VirtualPlacer()80 default_device_name_ = gpu_devices.begin()->second; in VirtualPlacer()
47 gpu_devices = config.list_physical_devices('GPU')48 if len(gpu_devices) == 1:50 config.set_logical_device_configuration(gpu_devices[0], [
54 gpu_devices = [57 self.devices = ['/cpu:0'] + gpu_devices