Searched refs:FindAttr (Results 1 – 8 of 8) sorted by relevance
/external/tensorflow/tensorflow/python/framework/ |
D | python_op_gen.cc | 174 const OpDef::AttrDef* attr = FindAttr(arg.number_attr(), op_def); in ArgTypeName() 197 const OpDef::AttrDef* attr = FindAttr(arg.type_attr(), op_def); in ArgTypeName() 215 const OpDef::AttrDef* attr = FindAttr(attr_name, op_def); in ArgTypeName() 451 const ApiDef::Attr* FindAttr(StringPiece name, const ApiDef& api_def) { in FindAttr() function 630 const auto& attr = *FindAttr(name, op_def_); in AddDocStringAttrs() 631 const auto& api_def_attr = *FindAttr(name, api_def_); in AddDocStringAttrs()
|
/external/tensorflow/tensorflow/core/framework/ |
D | graph_def_util.cc | 88 FindAttr(attr.first, *consumer_op_def) == nullptr) { in RemoveNewDefaultAttrsFromNodeDef() 90 FindAttr(attr.first, *producer_op_def); in RemoveNewDefaultAttrsFromNodeDef()
|
D | op_def_util.h | 43 const OpDef::AttrDef* FindAttr(StringPiece name, const OpDef& op_def);
|
D | op_def_util.cc | 146 const OpDef::AttrDef* FindAttr(StringPiece name, const OpDef& op_def) { in FindAttr() function 190 const OpDef::AttrDef* attr = FindAttr(arg.number_attr(), op_def); in ValidateArg() 215 const OpDef::AttrDef* attr = FindAttr(arg.type_attr(), op_def); in ValidateArg() 222 const OpDef::AttrDef* attr = FindAttr(arg.type_list_attr(), op_def); in ValidateArg()
|
D | fake_input.cc | 151 const OpDef::AttrDef* attr = FindAttr(arg_->type_attr(), *op_def_); in GetDataType()
|
D | op_def_builder.cc | 354 const OpDef::AttrDef* attr = FindAttr(type_or_attr, *op_def); in FinalizeInputOrOutput()
|
/external/tensorflow/tensorflow/compiler/tf2xla/ |
D | xla_op_registry.cc | 211 const OpDef::AttrDef& op_def_attr = *FindAttr(type_attr, *op_def); in RegisterCompilationKernels()
|
/external/tensorflow/tensorflow/python/eager/ |
D | python_eager_op_gen.cc | 483 const auto& attr = *FindAttr(attr_name, op_def_); in GetEagerFunctionSetup()
|