Searched refs:virtual_devices (Results 1 – 6 of 6) sorted by relevance
818 const auto& virtual_devices = gpu_options.experimental().virtual_devices(); in VerifyVirtualDeviceSettings() local819 CHECK(!virtual_devices.empty()); in VerifyVirtualDeviceSettings()825 if (num_gpus_to_use < virtual_devices.size()) { in VerifyVirtualDeviceSettings()829 num_gpus_to_use, " #virtual_devices: ", virtual_devices.size()); in VerifyVirtualDeviceSettings()832 visible_gpu_order.size() != virtual_devices.size()) { in VerifyVirtualDeviceSettings()837 " virtual_devices.size(): ", virtual_devices.size()); in VerifyVirtualDeviceSettings()839 if (valid_platform_gpu_ids.size() != virtual_devices.size()) { in VerifyVirtualDeviceSettings()844 " virtual_devices.size(): ", virtual_devices.size()); in VerifyVirtualDeviceSettings()1105 const auto& virtual_devices = gpu_options.experimental().virtual_devices(); in CreateDevices() local1106 if (!virtual_devices.empty()) { in CreateDevices()[all …]
80 auto virtual_devices = local83 virtual_devices->add_memory_limit_mb(mb);
164 auto virtual_devices = in TEST() local166 virtual_devices->add_memory_limit_mb(200); in TEST()167 virtual_devices->add_memory_limit_mb(200); in TEST()
63 name: "virtual_devices"
92 experimental=config_pb2.GPUOptions.Experimental(virtual_devices=[
104 // corresponding visible GPU (see "virtual_devices" below).123 // virtual_devices { memory_limit: 1GB memory_limit: 2GB }124 // virtual_devices {}136 repeated VirtualDevices virtual_devices = 1; field