Searched refs:api_def_attr (Results 1 – 6 of 6) sorted by relevance
| /external/tensorflow/tensorflow/cc/framework/ |
| D | cc_op_gen.cc | 594 const auto& api_def_attr(api_def.attr(i)); in OpInfo() local
595 CHECK_EQ(attr.name(), api_def_attr.name()); in OpInfo()
602 string attr_name = AvoidCPPKeywords(api_def_attr.rename_to()); in OpInfo()
605 if (!api_def_attr.description().empty()) { in OpInfo()
609 api_def_attr.description(), "\n"); in OpInfo()
611 if (api_def_attr.has_default_value()) { in OpInfo()
693 const auto& api_def_attr(api_def.attr(i)); in GetOpAttrStruct() local
698 !api_def_attr.has_default_value()) { in GetOpAttrStruct()
704 const string camel_case_name = ToCamelCase(api_def_attr.rename_to()); in GetOpAttrStruct()
712 if (!api_def_attr.description().empty()) { in GetOpAttrStruct()
[all …]
|
| /external/tensorflow/tensorflow/python/framework/ |
| D | python_op_gen.cc | 253 const auto& api_def_attr(api_def_.attr(i)); in Code() local
256 if (api_def_attr.has_default_value()) { in Code()
259 python_op_gen_internal::ParamNames(api_def_attr.name(), in Code()
260 api_def_attr.rename_to()), in Code()
263 TensorPBString(api_def_attr.default_value().tensor()), ", \"", in Code()
264 api_def_attr.rename_to(), "\")")); in Code()
267 for (const auto& pb : api_def_attr.default_value().list().tensor()) { in Code()
271 python_op_gen_internal::ParamNames(api_def_attr.name(), in Code()
272 api_def_attr.rename_to()), in Code()
274 api_def_attr.rename_to(), "\") for _pb in ", in Code()
[all …]
|
| D | python_op_gen_internal.cc | 688 const auto& api_def_attr = *FindAttr(name, api_def_); in AddDocStringAttrs() local
690 strings::StrCat(AvoidPythonReserved(api_def_attr.rename_to()), ": "); in AddDocStringAttrs()
714 if (api_def_attr.has_default_value()) { in AddDocStringAttrs()
743 if (api_def_attr.has_default_value()) { in AddDocStringAttrs()
746 AttrValueToPython(attr.type(), api_def_attr.default_value()), "`."); in AddDocStringAttrs()
748 if (!api_def_attr.description().empty()) { in AddDocStringAttrs()
749 AppendWithinWidth(&desc, api_def_attr.description(), in AddDocStringAttrs()
|
| /external/tensorflow/tensorflow/js/ops/ |
| D | ts_op_gen.cc | 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()
47 const ApiDef::Attr& api_def_attr; member
|
| /external/tensorflow/tensorflow/core/api_def/ |
| D | update_api_def.cc | 61 auto* api_def_attr = api_def->add_attr(); in FillBaseApiDef() local
62 api_def_attr->set_name(attr.name()); in FillBaseApiDef()
63 api_def_attr->set_description(attr.description()); in FillBaseApiDef()
|
| D | api_test.cc | 138 for (const auto& api_def_attr : api_def.attr()) { in TestAllApiDefAttributeNamesAreValid() local
141 if (api_def_attr.name() == op_attr.name()) { in TestAllApiDefAttributeNamesAreValid()
146 << "Attribute " << api_def_attr.name() << " (overwritten in api_def_" in TestAllApiDefAttributeNamesAreValid()
|