aboutsummaryrefslogtreecommitdiffhomepage
path: root/tensorflow/core/kernels/ops_testutil.cc
diff options
context:
space:
mode:
Diffstat (limited to 'tensorflow/core/kernels/ops_testutil.cc')
-rw-r--r--tensorflow/core/kernels/ops_testutil.cc18
1 files changed, 18 insertions, 0 deletions
diff --git a/tensorflow/core/kernels/ops_testutil.cc b/tensorflow/core/kernels/ops_testutil.cc
new file mode 100644
index 0000000000..7bea17b9e2
--- /dev/null
+++ b/tensorflow/core/kernels/ops_testutil.cc
@@ -0,0 +1,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