[res] Small network mocking the Epilogue (#6375)
author박천교/On-Device Lab(SR)/Engineer/삼성전자 <ch.bahk@samsung.com>
Thu, 8 Aug 2019 07:23:31 +0000 (16:23 +0900)
committer박세희/On-Device Lab(SR)/Principal Engineer/삼성전자 <saehie.park@samsung.com>
Thu, 8 Aug 2019 07:23:31 +0000 (16:23 +0900)
This commit adds TF_SMALL_NET_0014, a mock Epilogue network using ReLU
instead of Softmax

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

diff --git a/res/TensorFlowTests/TF_SMALL_NET_0014/test.info b/res/TensorFlowTests/TF_SMALL_NET_0014/test.info
new file mode 100644 (file)
index 0000000..d3ea85a
--- /dev/null
@@ -0,0 +1,2 @@
+input,  placeholder:0,  TF_FLOAT,   [2, 1, 1, 3]
+output, reshape_2:0,    TF_FLOAT,   [2, 3]
diff --git a/res/TensorFlowTests/TF_SMALL_NET_0014/test.pbtxt b/res/TensorFlowTests/TF_SMALL_NET_0014/test.pbtxt
new file mode 100644 (file)
index 0000000..d4bc8e6
--- /dev/null
@@ -0,0 +1,106 @@
+# Mock of the Epilogue, using ReLU instead of Softmax
+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: 1 }
+        dim { size: 3 }
+      }
+    }
+  }
+}
+node {
+  name: "squeeze"
+  op: "Squeeze"
+  input: "placeholder"
+  attr {
+    key: "T"
+    value { type: DT_FLOAT }
+  }
+  attr {
+    key: "squeeze_dims"
+    value {
+      list { i: 1 i: 2 }
+    }
+  }
+}
+node {
+  name: "Reshape/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: 3
+      }
+    }
+  }
+}
+node {
+  name: "reshape_1"
+  op: "Reshape"
+  input: "squeeze"
+  input: "Reshape/shape"
+  attr {
+    key: "T"
+    value { type: DT_FLOAT }
+  }
+  attr {
+    key: "Tshape"
+    value { type: DT_INT32 }
+  }
+}
+node {
+  name: "relu"
+  op: "Relu"
+  input: "reshape_1"
+  attr {
+    key: "T"
+    value { type: DT_FLOAT }
+  }
+}
+node {
+  name: "shape"
+  op: "Shape"
+  input: "squeeze"
+  attr {
+    key: "T"
+    value { type: DT_FLOAT }
+  }
+  attr {
+    key: "out_type"
+    value { type: DT_INT32 }
+  }
+}
+node {
+  name: "reshape_2"
+  op: "Reshape"
+  input: "relu"
+  input: "shape"
+  attr {
+    key: "T"
+    value { type: DT_FLOAT }
+  }
+  attr {
+    key: "Tshape"
+    value { type: DT_INT32 }
+  }
+}