aboutsummaryrefslogtreecommitdiffhomepage
path: root/tensorflow/python/framework/python_op_gen.h
blob: 488f7431e0fb41f5314869209c82836a06b44c1e (plain)
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
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_