Home
last modified time | relevance | path

Searched refs:op_registration (Results 1 – 1 of 1) sorted by relevance

/external/tensorflow/tensorflow/compiler/tf2xla/
Dxla_op_registry.cc202 for (auto& op_registration : op_registrations) { in RegisterCompilationKernels() local
203 if (op_registration->has_device_whitelist) { in RegisterCompilationKernels()
204 whitelisted_backend.insert(op_registration->device_whitelist.begin(), in RegisterCompilationKernels()
205 op_registration->device_whitelist.end()); in RegisterCompilationKernels()
209 for (auto& op_registration : op_registrations) { in RegisterCompilationKernels() local
229 for (const auto& constraint : op_registration->type_constraints) { in RegisterCompilationKernels()
239 if (op_registration->has_device_whitelist && in RegisterCompilationKernels()
240 op_registration->device_whitelist.find(backend.first) == in RegisterCompilationKernels()
241 op_registration->device_whitelist.end()) { in RegisterCompilationKernels()
247 if (!op_registration->has_device_whitelist && in RegisterCompilationKernels()
[all …]