Searched refs:device_whitelist (Results 1 – 3 of 3) sorted by relevance
87 device_whitelist=None, argument94 self.device_whitelist = device_whitelist135 if self.device_whitelist:136 flags['BUFFET_DEVICE_WHITELIST'] = ','.join(self.device_whitelist)
87 for (const auto& device : x.device_whitelist) { in IsCompatible()88 if (y.device_whitelist.count(device) != 0) { in IsCompatible()204 whitelisted_backend.insert(op_registration->device_whitelist.begin(), in RegisterCompilationKernels()205 op_registration->device_whitelist.end()); in RegisterCompilationKernels()240 op_registration->device_whitelist.find(backend.first) == in RegisterCompilationKernels()241 op_registration->device_whitelist.end()) { in RegisterCompilationKernels()475 registration_->device_whitelist.emplace(device); in Device()483 registration_->device_whitelist.emplace(device); in Device()
225 std::unordered_set<string> device_whitelist; member