diff options
| author |  Anna R <annarev@google.com> | 2017-11-14 15:00:48 -0800 |
|---|---|---|
| committer |  TensorFlower Gardener <gardener@tensorflow.org> | 2017-11-14 15:08:52 -0800 |
| commit | f89cffd37c88e4d9fa0ee3ac191e6f5fd5c005c8 (patch) | |
| tree | c2e91df197ac83a606820c182d7af3c086128b2f /tensorflow/core/framework/op_def_util.h | |
| parent | bc0b26046fa729612b0017815b72a5faa4890e86 (diff) | |
Internal change.
PiperOrigin-RevId: 175740778
Diffstat (limited to 'tensorflow/core/framework/op_def_util.h')
| -rw-r--r-- | tensorflow/core/framework/op_def_util.h | 4 |
1 files changed, 4 insertions, 0 deletions
diff --git a/tensorflow/core/framework/op_def_util.h b/tensorflow/core/framework/op_def_util.h index c329e4627c..f9661dcedd 100644 --- a/tensorflow/core/framework/op_def_util.h +++ b/tensorflow/core/framework/op_def_util.h @@ -43,6 +43,10 @@ Status ValidateAttrValue(const AttrValue& attr_value, const OpDef::AttrDef* FindAttr(StringPiece name, const OpDef& op_def); OpDef::AttrDef* FindAttrMutable(StringPiece name, OpDef* op_def); +// Searches op_def for input argument with the indicated name. +// Returns nullptr if no such attr is found. +const OpDef::ArgDef* FindInputArg(StringPiece name, const OpDef& op_def); + // Produce a human-readable version of an op_def that is more concise // than a text-format proto. Excludes descriptions. string SummarizeOpDef(const OpDef& op_def); |