blob: 7bea17b9e28eea3dbb875a14d4d60fe9d902f88f (
plain)
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
|
#include "tensorflow/core/kernels/ops_testutil.h"
namespace tensorflow {
namespace test {
NodeDef Node(const string& name, const string& op,
const std::vector<string>& inputs) {
NodeDef def;
def.set_name(name);
def.set_op(op);
for (const string& s : inputs) {
def.add_input(s);
}
return def;
}
} // namespace test
} // namespace tensorflow
|