diff options
author | Nick Kreeger <kreeger@google.com> | 2018-08-17 14:26:35 -0700 |
---|---|---|
committer | TensorFlower Gardener <gardener@tensorflow.org> | 2018-08-17 14:31:18 -0700 |
commit | c4ccab7a95835ec12af16938ba0c17cc7c1f17e0 (patch) | |
tree | a1144890c0eaeda823fac1272a6326939ec97491 /tensorflow/python/framework | |
parent | f67c02ce213e0420d91de61f896f1716a4943e7e (diff) |
Move FindInputArg() for ApiDef::Arg to op_def_util.cc and cleanup Python/c++ references.
This method is currently duplicated in Python code and will be used for TypeScript generation.
PiperOrigin-RevId: 209207139
Diffstat (limited to 'tensorflow/python/framework')
-rw-r--r-- | tensorflow/python/framework/python_op_gen.cc | 9 | ||||
-rw-r--r-- | tensorflow/python/framework/python_op_gen_internal.cc | 9 |
2 files changed, 0 insertions, 18 deletions
diff --git a/tensorflow/python/framework/python_op_gen.cc b/tensorflow/python/framework/python_op_gen.cc index 76d4c2017c..2022fbcbaa 100644 --- a/tensorflow/python/framework/python_op_gen.cc +++ b/tensorflow/python/framework/python_op_gen.cc @@ -102,15 +102,6 @@ string TensorPBString(const TensorProto& pb) { return strings::StrCat("\"\"\"", ProtoShortDebugString(pb), "\"\"\""); } -const ApiDef::Arg* FindInputArg(StringPiece name, const ApiDef& api_def) { - for (int i = 0; i < api_def.in_arg_size(); ++i) { - if (api_def.in_arg(i).name() == name) { - return &api_def.in_arg(i); - } - } - return nullptr; -} - class GenEagerPythonOp : public python_op_gen_internal::GenPythonOp { public: GenEagerPythonOp(const OpDef& op_def, const ApiDef& api_def, diff --git a/tensorflow/python/framework/python_op_gen_internal.cc b/tensorflow/python/framework/python_op_gen_internal.cc index 031b4a384e..f2270342b0 100644 --- a/tensorflow/python/framework/python_op_gen_internal.cc +++ b/tensorflow/python/framework/python_op_gen_internal.cc @@ -483,15 +483,6 @@ const ApiDef::Attr* FindAttr(StringPiece name, const ApiDef& api_def) { return nullptr; } -const ApiDef::Arg* FindInputArg(StringPiece name, const ApiDef& api_def) { - for (int i = 0; i < api_def.in_arg_size(); ++i) { - if (api_def.in_arg(i).name() == name) { - return &api_def.in_arg(i); - } - } - return nullptr; -} - GenPythonOp::GenPythonOp(const OpDef& op_def, const ApiDef& api_def, const string& function_name) : op_def_(op_def), |