aboutsummaryrefslogtreecommitdiffhomepage
path: root/tensorflow/cc/ops/cc_op_gen.h
blob: 1a9474ec874284f6cb8ebb1b3a0403a687f441a5 (plain)
1
2
3
4
5
6
7
8
9
10
11
12
13
14
#ifndef TENSORFLOW_CC_OPS_CC_OP_GEN_H_
#define TENSORFLOW_CC_OPS_CC_OP_GEN_H_

#include "tensorflow/core/framework/op_def.pb.h"

namespace tensorflow {

// Result is written to files dot_h and dot_cc.
void WriteCCOps(const OpList& ops, const std::string& dot_h_fname,
                const std::string& dot_cc_fname);

}  // namespace tensorflow

#endif  // TENSORFLOW_CC_OPS_CC_OP_GEN_H_