Searched refs:op_def_attr (Results 1 – 2 of 2) sorted by relevance
43 OpAttrs(const OpDef::AttrDef& op_def_attr, const ApiDef::Attr& api_def_attr) in OpAttrs()44 : op_def_attr(op_def_attr), api_def_attr(api_def_attr) {} in OpAttrs()46 const OpDef::AttrDef& op_def_attr; member63 string InputForAttr(const OpDef::AttrDef& op_def_attr);155 string GenTypeScriptOp::InputForAttr(const OpDef::AttrDef& op_def_attr) { in InputForAttr() argument157 auto arg_list = attr_arg_idx_map_.find(op_def_attr.name()); in InputForAttr()210 if (attr.op_def_attr.type() == "type") { in AddOpAttrs()213 attr.op_def_attr.name(), "', ", in AddOpAttrs()214 InputForAttr(attr.op_def_attr), ")"); in AddOpAttrs()215 } else if (attr.op_def_attr.type() == "int") { in AddOpAttrs()[all …]
280 const OpDef::AttrDef& op_def_attr = *FindAttr(type_attr, *op_def); in RegisterCompilationKernels() local282 op_def_attr.has_allowed_values() in RegisterCompilationKernels()283 ? &op_def_attr.allowed_values().list().type() in RegisterCompilationKernels()