Searched refs:virtual_devices (Results 1 – 8 of 8) sorted by relevance
117 bool UsesVirtualDevice(const Optional<DictionaryAttr>& virtual_devices, in UsesVirtualDevice() argument119 if (!virtual_devices.hasValue()) return false; in UsesVirtualDevice()125 if (virtual_devices.getValue().get(op_device.getValue())) in UsesVirtualDevice()
348 options.experimental().virtual_devices().Get(i).priority_size(); in GetPriority()353 options.experimental().virtual_devices().Get(i).priority().Get(id); in GetPriority()831 const auto& virtual_devices = gpu_options.experimental().virtual_devices(); in VerifyVirtualDeviceSettings() local832 CHECK(!virtual_devices.empty()); in VerifyVirtualDeviceSettings()838 if (num_gpus_to_use < virtual_devices.size()) { in VerifyVirtualDeviceSettings()842 num_gpus_to_use, " #virtual_devices: ", virtual_devices.size()); in VerifyVirtualDeviceSettings()845 visible_gpu_order.size() != virtual_devices.size()) { in VerifyVirtualDeviceSettings()850 " virtual_devices.size(): ", virtual_devices.size()); in VerifyVirtualDeviceSettings()852 if (valid_platform_gpu_ids.size() != virtual_devices.size()) { in VerifyVirtualDeviceSettings()857 " virtual_devices.size(): ", virtual_devices.size()); in VerifyVirtualDeviceSettings()[all …]
83 auto virtual_devices = local86 virtual_devices->add_memory_limit_mb(mb);90 virtual_devices->add_priority(p);
91 auto virtual_devices = in TEST() local93 virtual_devices->add_memory_limit_mb(200); in TEST()94 virtual_devices->add_memory_limit_mb(200); in TEST()
1018 virtual_devices = self._virtual_device_map.get(dev)1019 if virtual_devices is None:1022 config.device_count[dev.device_type] += len(virtual_devices)1050 virtual_devices = []1076 virtual_devices.append(1082 if not virtual_devices and memory_growths:1093 virtual_devices=virtual_devices))1493 def set_logical_device_configuration(self, dev, virtual_devices): argument1501 for vdev in virtual_devices:1509 for vdev in virtual_devices:[all …]
63 name: "virtual_devices"
92 experimental=config_pb2.GPUOptions.Experimental(virtual_devices=[
106 // corresponding visible GPU (see "virtual_devices" below).137 // virtual_devices { memory_limit: 1GB memory_limit: 2GB }138 // virtual_devices {}150 repeated VirtualDevices virtual_devices = 1; field