Searched refs:type_constraints (Results 1 – 2 of 2) sorted by relevance
229 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() local276 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()
220 std::unordered_map<string, std::set<DataType>> type_constraints; member