[res] Introduce UNIT_Squeeze_003 test (#7039)
author남궁석/On-Device Lab(SR)/Engineer/삼성전자 <sk.namkoong@samsung.com>
Fri, 30 Aug 2019 00:04:20 +0000 (09:04 +0900)
committer박종현/On-Device Lab(SR)/Staff Engineer/삼성전자 <jh1302.park@samsung.com>
Fri, 30 Aug 2019 00:04:20 +0000 (09:04 +0900)
This test is for checking that `Squeeze` operation handle multiple negative
squeeze_dims or not

Signed-off-by: Seok NamKoong <sk.namkoong@samsung.com>
res/TensorFlowTests/UNIT_Squeeze_003/test.info [new file with mode: 0644]
res/TensorFlowTests/UNIT_Squeeze_003/test.pbtxt [new file with mode: 0644]

diff --git a/res/TensorFlowTests/UNIT_Squeeze_003/test.info b/res/TensorFlowTests/UNIT_Squeeze_003/test.info
new file mode 100644 (file)
index 0000000..7cc93f3
--- /dev/null
@@ -0,0 +1,2 @@
+input,  Placeholder:0, TF_FLOAT, [2, 1, 3, 1]
+output, Squeeze:0,     TF_FLOAT, [2, 3]
diff --git a/res/TensorFlowTests/UNIT_Squeeze_003/test.pbtxt b/res/TensorFlowTests/UNIT_Squeeze_003/test.pbtxt
new file mode 100644 (file)
index 0000000..1c760c8
--- /dev/null
@@ -0,0 +1,35 @@
+# Network with Squeeze that have multiple negative squeeze dimensions
+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 i: -3 }
+    }
+  }
+}