[res] Add small network test case for ResolveReshapeWildcardDim (#6020)
author박천교/On-Device Lab(SR)/Engineer/삼성전자 <ch.bahk@samsung.com>
Thu, 1 Aug 2019 01:35:26 +0000 (10:35 +0900)
committer박종현/On-Device Lab(SR)/Staff Engineer/삼성전자 <jh1302.park@samsung.com>
Thu, 1 Aug 2019 01:35:26 +0000 (10:35 +0900)
* [res] Add small network test case for ResolveReshapeWildcardDim

This commit adds small network test case for ResolveReshapeWildcardDim.

Signed-off-by: Cheongyo Bahk <ch.bahk@samsung.com>
* Edit brief description

res/TensorFlowTests/TF_SMALL_NET_0012/test.info [new file with mode: 0644]
res/TensorFlowTests/TF_SMALL_NET_0012/test.pbtxt [new file with mode: 0644]

diff --git a/res/TensorFlowTests/TF_SMALL_NET_0012/test.info b/res/TensorFlowTests/TF_SMALL_NET_0012/test.info
new file mode 100644 (file)
index 0000000..b210407
--- /dev/null
@@ -0,0 +1,2 @@
+input,  placeholder:0, TF_FLOAT, [6]
+output, reshape:0,     TF_FLOAT, [3, 2]
diff --git a/res/TensorFlowTests/TF_SMALL_NET_0012/test.pbtxt b/res/TensorFlowTests/TF_SMALL_NET_0012/test.pbtxt
new file mode 100644 (file)
index 0000000..387dc92
--- /dev/null
@@ -0,0 +1,52 @@
+# Network with dynamic reshape which has resolvable wildcard dimension
+node {
+  name: "placeholder"
+  op: "Placeholder"
+  attr {
+    key: "dtype"
+    value { type: DT_FLOAT }
+  }
+  attr {
+    key: "shape"
+    value {
+      shape {
+        dim { size: 6 }
+      }
+    }
+  }
+}
+node {
+  name: "shape"
+  op: "Const"
+  attr {
+    key: "dtype"
+    value { type: DT_INT32 }
+  }
+  attr {
+    key: "value"
+    value {
+      tensor {
+        dtype: DT_INT32
+        tensor_shape {
+          dim { size: 2 }
+        }
+        int_val: -1
+        int_val: 2
+      }
+    }
+  }
+}
+node {
+  name: "reshape"
+  op: "Reshape"
+  input: "placeholder"
+  input: "shape"
+  attr {
+    key: "T"
+    value { type: DT_FLOAT }
+  }
+  attr {
+    key: "Tshape"
+    value { type: DT_INT32 }
+  }
+}