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.cc213 for (auto& op_registration : op_registrations) { in RegisterCompilationKernels() local
214 if (op_registration->has_device_allowlist) { in RegisterCompilationKernels()
215 allowlisted_backend.insert(op_registration->device_allowlist.begin(), in RegisterCompilationKernels()
216 op_registration->device_allowlist.end()); in RegisterCompilationKernels()
220 for (auto& op_registration : op_registrations) { in RegisterCompilationKernels() local
240 for (const auto& constraint : op_registration->type_constraints) { in RegisterCompilationKernels()
250 if (op_registration->has_device_allowlist && in RegisterCompilationKernels()
251 op_registration->device_allowlist.find(backend.first) == in RegisterCompilationKernels()
252 op_registration->device_allowlist.end()) { in RegisterCompilationKernels()
258 if (!op_registration->has_device_allowlist && in RegisterCompilationKernels()
[all …]