Searched refs:api_def_attr (Results 1 – 6 of 6) sorted by relevance
| /external/tensorflow/tensorflow/cc/framework/ |
| D | cc_op_gen.cc | 616 const auto& api_def_attr(api_def.attr(i)); in OpInfo() local
617 CHECK_EQ(attr.name(), api_def_attr.name()); in OpInfo()
624 string attr_name = AvoidCPPKeywords(api_def_attr.rename_to()); in OpInfo()
627 if (!api_def_attr.description().empty()) { in OpInfo()
631 api_def_attr.description(), "\n"); in OpInfo()
633 if (api_def_attr.has_default_value()) { in OpInfo()
715 const auto& api_def_attr(api_def.attr(i)); in GetOpAttrStruct() local
720 !api_def_attr.has_default_value()) { in GetOpAttrStruct()
726 const string camel_case_name = ToCamelCase(api_def_attr.rename_to()); in GetOpAttrStruct()
734 if (!api_def_attr.description().empty()) { in GetOpAttrStruct()
[all …]
|
| /external/tensorflow/tensorflow/python/framework/ |
| D | python_op_gen.cc | 294 const auto& api_def_attr(api_def_.attr(i)); in Code() local
297 if (api_def_attr.has_default_value()) { in Code()
300 python_op_gen_internal::ParamNames(api_def_attr.name(), in Code()
301 api_def_attr.rename_to()), in Code()
304 TensorPBString(api_def_attr.default_value().tensor()), ", \"", in Code()
305 api_def_attr.rename_to(), "\")")); in Code()
308 for (const auto& pb : api_def_attr.default_value().list().tensor()) { in Code()
312 python_op_gen_internal::ParamNames(api_def_attr.name(), in Code()
313 api_def_attr.rename_to()), in Code()
315 api_def_attr.rename_to(), "\") for _pb in ", in Code()
[all …]
|
| D | python_op_gen_internal.cc | 709 const auto& api_def_attr = *FindAttr(name, api_def_); in AddDocStringAttrs() local
711 strings::StrCat(AvoidPythonReserved(api_def_attr.rename_to()), ": "); in AddDocStringAttrs()
735 if (api_def_attr.has_default_value()) { in AddDocStringAttrs()
764 if (api_def_attr.has_default_value()) { in AddDocStringAttrs()
767 AttrValueToPython(attr.type(), api_def_attr.default_value()), "`."); in AddDocStringAttrs()
769 if (!api_def_attr.description().empty()) { in AddDocStringAttrs()
770 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 | 147 for (const auto& api_def_attr : api_def.attr()) { in TestAllApiDefAttributeNamesAreValid() local
150 if (api_def_attr.name() == op_attr.name()) { in TestAllApiDefAttributeNamesAreValid()
155 << "Attribute " << api_def_attr.name() << " (overwritten in api_def_" in TestAllApiDefAttributeNamesAreValid()
|