[res] Add negative squeeze dimension test case (#6262)
author박천교/On-Device Lab(SR)/Engineer/삼성전자 <ch.bahk@samsung.com>
Tue, 6 Aug 2019 09:33:30 +0000 (18:33 +0900)
committer박세희/On-Device Lab(SR)/Principal Engineer/삼성전자 <saehie.park@samsung.com>
Tue, 6 Aug 2019 09:33:30 +0000 (18:33 +0900)
This commit add Squeeze network test case with negative squeeze dimension

Signed-off-by: Cheongyo Bahk <ch.bahk@samsung.com>
res/TensorFlowTests/Squeeze_002/test.info [new file with mode: 0644]
res/TensorFlowTests/Squeeze_002/test.pbtxt [new file with mode: 0644]

diff --git a/res/TensorFlowTests/Squeeze_002/test.info b/res/TensorFlowTests/Squeeze_002/test.info
new file mode 100644 (file)
index 0000000..401e102
--- /dev/null
@@ -0,0 +1,2 @@
+input,  Placeholder:0, TF_FLOAT, [2, 1, 3, 1]
+output, Squeeze:0,     TF_FLOAT, [2, 1, 3]
diff --git a/res/TensorFlowTests/Squeeze_002/test.pbtxt b/res/TensorFlowTests/Squeeze_002/test.pbtxt
new file mode 100644 (file)
index 0000000..06d0901
--- /dev/null
@@ -0,0 +1,35 @@
+# Network with Squeeze that has negative 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 }
+    }
+  }
+}