Searched refs:virtual_devices (Results 1 – 7 of 7) sorted by relevance
916 virtual_devices = self._virtual_device_map.get(dev)917 if virtual_devices is None:920 config.device_count[dev.device_type] += len(virtual_devices)948 virtual_devices = []967 virtual_devices.append(973 if not virtual_devices and memory_growths:984 virtual_devices=virtual_devices))1284 def set_logical_device_configuration(self, dev, virtual_devices): argument1292 for vdev in virtual_devices:1297 for vdev in virtual_devices:[all …]
90 auto virtual_devices = in TEST() local92 virtual_devices->add_memory_limit_mb(200); in TEST()93 virtual_devices->add_memory_limit_mb(200); in TEST()
773 const auto& virtual_devices = gpu_options.experimental().virtual_devices(); in VerifyVirtualDeviceSettings() local774 CHECK(!virtual_devices.empty()); in VerifyVirtualDeviceSettings()780 if (num_gpus_to_use < virtual_devices.size()) { in VerifyVirtualDeviceSettings()784 num_gpus_to_use, " #virtual_devices: ", virtual_devices.size()); in VerifyVirtualDeviceSettings()787 visible_gpu_order.size() != virtual_devices.size()) { in VerifyVirtualDeviceSettings()792 " virtual_devices.size(): ", virtual_devices.size()); in VerifyVirtualDeviceSettings()794 if (valid_platform_gpu_ids.size() != virtual_devices.size()) { in VerifyVirtualDeviceSettings()799 " virtual_devices.size(): ", virtual_devices.size()); in VerifyVirtualDeviceSettings()1122 const auto& virtual_devices = gpu_options.experimental().virtual_devices(); in CreateDevices() local1123 if (!virtual_devices.empty()) { in CreateDevices()[all …]
79 auto virtual_devices = local82 virtual_devices->add_memory_limit_mb(mb);
63 name: "virtual_devices"
92 experimental=config_pb2.GPUOptions.Experimental(virtual_devices=[
106 // corresponding visible GPU (see "virtual_devices" below).125 // virtual_devices { memory_limit: 1GB memory_limit: 2GB }126 // virtual_devices {}138 repeated VirtualDevices virtual_devices = 1; field