diff options
Diffstat (limited to 'tensorflow/python/framework/python_op_gen.h')
| -rw-r--r-- | tensorflow/python/framework/python_op_gen.h | 17 |
1 files changed, 17 insertions, 0 deletions
diff --git a/tensorflow/python/framework/python_op_gen.h b/tensorflow/python/framework/python_op_gen.h new file mode 100644 index 0000000000..488f7431e0 --- /dev/null +++ b/tensorflow/python/framework/python_op_gen.h @@ -0,0 +1,17 @@ +#ifndef TENSORFLOW_PYTHON_FRAMEWORK_PYTHON_OP_GEN_H_ +#define TENSORFLOW_PYTHON_FRAMEWORK_PYTHON_OP_GEN_H_ + +#include <string> +#include "tensorflow/core/framework/op_def.pb.h" +#include "tensorflow/core/platform/port.h" + +namespace tensorflow { + +// Result is printed to stdout. hidden_ops should be a comma-separated +// list of Op names that should get a leading _ in the output. +void PrintPythonOps(const OpList& ops, const string& hidden_ops, + bool require_shapes); + +} // namespace tensorflow + +#endif // TENSORFLOW_PYTHON_FRAMEWORK_PYTHON_OP_GEN_H_ |