1 files changed, 35 insertions, 0 deletions
diff --git a/res/TensorFlowTests/UNIT_Squeeze_001/test.pbtxt b/res/TensorFlowTests/UNIT_Squeeze_001/test.pbtxt
new file mode 100644
index 000000000..641300c10
--- /dev/null
+++ b/res/TensorFlowTests/UNIT_Squeeze_001/test.pbtxt
@@ -0,0 +1,35 @@
+# Network with Squeeze that has selective squeeze dimension
+node {
+  name: "Placeholder"
+  op: "Placeholder"
+  attr {
+    key: "dtype"
+    value { type: DT_FLOAT }
+  }
+  attr {
+    key: "shape"
+    value {
+      shape {
+        dim { size: 2 }
+        dim { size: 1 }
+        dim { size: 3 }
+        dim { size: 1 }
+      }
+    }
+  }
+}
+node {
+  name: "Squeeze"
+  op: "Squeeze"
+  input: "Placeholder"
+  attr {
+    key: "T"
+    value { type: DT_FLOAT }
+  }
+  attr {
+    key: "squeeze_dims"
+    value {
+      list { i: 1 }
+    }
+  }
+}