aboutsummaryrefslogtreecommitdiffhomepage
path: root/tensorflow/core/framework/op_gen_lib.h
diff options
context:
space:
mode:
Diffstat (limited to 'tensorflow/core/framework/op_gen_lib.h')
-rw-r--r--tensorflow/core/framework/op_gen_lib.h24
1 files changed, 24 insertions, 0 deletions
diff --git a/tensorflow/core/framework/op_gen_lib.h b/tensorflow/core/framework/op_gen_lib.h
new file mode 100644
index 0000000000..9890f1bcec
--- /dev/null
+++ b/tensorflow/core/framework/op_gen_lib.h
@@ -0,0 +1,24 @@
+#ifndef TENSORFLOW_FRAMEWORK_OP_GEN_LIB_H_
+#define TENSORFLOW_FRAMEWORK_OP_GEN_LIB_H_
+
+#include <string>
+#include "tensorflow/core/lib/core/stringpiece.h"
+
+namespace tensorflow {
+
+inline string Spaces(int n) { return string(n, ' '); }
+
+// Wrap prefix + str to be at most width characters, indenting every line
+// after the first by prefix.size() spaces. Intended use case is something
+// like prefix = " Foo(" and str is a list of arguments (terminated by a ")").
+// TODO(josh11b): Option to wrap on ", " instead of " " when possible.
+string WordWrap(StringPiece prefix, StringPiece str, int width);
+
+// Looks for an "=" at the beginning of *description. If found, strips it off
+// (and any following spaces) from *description and return true. Otherwise
+// returns false.
+bool ConsumeEquals(StringPiece* description);
+
+} // namespace tensorflow
+
+#endif // TENSORFLOW_FRAMEWORK_OP_GEN_LIB_H_