Home
last modified time | relevance | path

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

/external/tensorflow/tensorflow/compiler/tf2xla/
Dxla_op_registry.cc229 for (const auto& constraint : op_registration->type_constraints) { in RegisterCompilationKernels()
274 op_registration->type_constraints.find(type_attr); in RegisterCompilationKernels()
275 const std::set<DataType>* type_constraints = in RegisterCompilationKernels() local
276 constraint_it != op_registration->type_constraints.end() in RegisterCompilationKernels()
288 if (type_constraints != nullptr && in RegisterCompilationKernels()
289 type_constraints->find(dtype) == type_constraints->end()) { in RegisterCompilationKernels()
505 registration_->type_constraints[string(attr_name)]; in TypeConstraint()
513 registration_->type_constraints[string(attr_name)]; in TypeConstraint()
Dxla_op_registry.h220 std::unordered_map<string, std::set<DataType>> type_constraints; member