Searched refs:type_constraints (Results 1 – 2 of 2) sorted by relevance
240 for (const auto& constraint : op_registration->type_constraints) { in RegisterCompilationKernels()286 op_registration->type_constraints.find(type_attr); in RegisterCompilationKernels()287 const std::set<DataType>* type_constraints = in RegisterCompilationKernels() local288 constraint_it != op_registration->type_constraints.end() in RegisterCompilationKernels()300 if (type_constraints != nullptr && in RegisterCompilationKernels()301 type_constraints->find(dtype) == type_constraints->end()) { in RegisterCompilationKernels()531 registration_->type_constraints[string(attr_name)]; in TypeConstraint()539 registration_->type_constraints[string(attr_name)]; in TypeConstraint()
268 std::unordered_map<string, std::set<DataType>> type_constraints; member