Searched refs:FindAttr (Results 1 – 8 of 8) sorted by relevance
| /external/tensorflow/tensorflow/python/framework/ |
| D | python_op_gen_internal.cc | 205 const OpDef::AttrDef* attr = FindAttr(arg.number_attr(), op_def); in ArgTypeName()
228 const OpDef::AttrDef* attr = FindAttr(arg.type_attr(), op_def); in ArgTypeName()
246 const OpDef::AttrDef* attr = FindAttr(attr_name, op_def); in ArgTypeName()
487 const ApiDef::Attr* FindAttr(StringPiece name, const ApiDef& api_def) { in FindAttr() function
687 const auto& attr = *FindAttr(name, op_def_); in AddDocStringAttrs()
688 const auto& api_def_attr = *FindAttr(name, api_def_); in AddDocStringAttrs()
|
| D | python_op_gen.cc | 478 const auto& attr = *FindAttr(attr_name, op_def_); in GetEagerFunctionSetup()
|
| /external/tensorflow/tensorflow/core/framework/ |
| D | graph_def_util.cc | 99 FindAttr(attr.first, *consumer_op_def) == nullptr) { in RemoveNewDefaultAttrsFromNodeDef()
101 FindAttr(attr.first, *producer_op_def); in RemoveNewDefaultAttrsFromNodeDef()
|
| D | op_def_util.h | 44 const OpDef::AttrDef* FindAttr(StringPiece name, const OpDef& op_def);
|
| D | op_def_util.cc | 150 const OpDef::AttrDef* FindAttr(StringPiece name, const OpDef& op_def) { in FindAttr() function
203 const OpDef::AttrDef* attr = FindAttr(arg.number_attr(), op_def); in ValidateArg()
228 const OpDef::AttrDef* attr = FindAttr(arg.type_attr(), op_def); in ValidateArg()
235 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 | 367 const OpDef::AttrDef* attr = FindAttr(type_or_attr, *op_def); in FinalizeInputOrOutput()
|
| /external/tensorflow/tensorflow/compiler/tf2xla/ |
| D | xla_op_registry.cc | 268 const OpDef::AttrDef& op_def_attr = *FindAttr(type_attr, *op_def); in RegisterCompilationKernels()
|