3 files changed, 77 insertions, 0 deletions
diff --git a/res/TensorFlowTests/UNIT_CustomOp_000/customop.conf b/res/TensorFlowTests/UNIT_CustomOp_000/customop.conf
new file mode 100644
index 000000000..08ba0f09a
--- /dev/null
+++ b/res/TensorFlowTests/UNIT_CustomOp_000/customop.conf
@@ -0,0 +1,22 @@
+# defining a list of custom ops for this compilation
+custom_op {
+  name: "my/customOp/000"
+  op: "new_custom_op"
+  attr {
+    key: "dtype"
+    value {
+      type: DT_FLOAT
+    }
+  }
+  attr {
+    key: "output_shape"
+    value {
+      shape {
+        dim { size: 1 }
+        dim { size: 2 }
+        dim { size: 1 }
+        dim { size: 2 }
+      }
+    }
+  }
+}
diff --git a/res/TensorFlowTests/UNIT_CustomOp_000/test.info b/res/TensorFlowTests/UNIT_CustomOp_000/test.info
new file mode 100644
index 000000000..e3e2b176e
--- /dev/null
+++ b/res/TensorFlowTests/UNIT_CustomOp_000/test.info
@@ -0,0 +1,2 @@
+input,  Placeholder:0,      TF_FLOAT,   [1, 2, 1, 2]
+output, output/relu:0,      TF_FLOAT,   [1, 2, 1, 2]
diff --git a/res/TensorFlowTests/UNIT_CustomOp_000/test.pbtxt b/res/TensorFlowTests/UNIT_CustomOp_000/test.pbtxt
new file mode 100644
index 000000000..373f5b845
--- /dev/null
+++ b/res/TensorFlowTests/UNIT_CustomOp_000/test.pbtxt
@@ -0,0 +1,53 @@
+node {
+  name: "Placeholder"
+  op: "Placeholder"
+  attr {
+    key: "dtype" value { type: DT_FLOAT } }
+  attr {
+    key: "shape"
+    value {
+      shape {
+        dim { size: 1 }
+        dim { size: 2 }
+        dim { size: 1 }
+        dim { size: 2 }
+      }
+    }
+  }
+}
+node {
+  name: "my/customOp/000"
+  op: "new_custom_op"
+  input: "Placeholder"
+  attr {
+    key: "T"
+    value { type: DT_FLOAT }
+  }
+  attr {
+    key: "my_string"
+    value { s: "Hello World" }
+  }
+  attr {
+    key: "my_float"
+    value { f: 0.001 }
+  }
+  attr {
+    key: "my_int"
+    value { i: 111 }
+  }
+  attr {
+    key: "my_bool"
+    value { b: false }
+  }
+}
+node {
+  name: "output/relu"
+  op: "Relu"
+  input: "my/customOp/000"
+  attr {
+    key: "T"
+    value {
+      type: DT_FLOAT
+    }
+  }
+}