Searched refs:type_constraints (Results 1 – 5 of 5) sorted by relevance
148 reg.type_constraints["foo"] = DT_FLOAT; in TEST()149 reg.type_constraints["bar"] = DT_INT32; in TEST()172 reg1.type_constraints["foo"] = DT_FLOAT; in TEST()173 reg2.type_constraints["foo"] = DT_INT32; in TEST()196 reg.type_constraints["foo"] = DT_FLOAT; in TEST()197 reg.type_constraints["bar"] = DT_INT32; in TEST()
266 llvm::StringMap<DataType> type_constraints; member
307 kernel_info.type_constraints); in CreateKernel()
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