summaryrefslogtreecommitdiff
path: root/res/TensorFlowTests
diff options
context:
space:
mode:
author박천교/On-Device Lab(SR)/Engineer/삼성전자 <ch.bahk@samsung.com>2019-08-08 16:23:31 +0900
committer박세희/On-Device Lab(SR)/Principal Engineer/삼성전자 <saehie.park@samsung.com>2019-08-08 16:23:31 +0900
commit89b50c6bfaa9c54c6dde445894843cd433dea6a9 (patch)
tree76ebc173b4e4fe5e793b287bdc4fb1f7f9264361 /res/TensorFlowTests
parente8019fc78435b2f4e88d33f1f31522f8271c545f (diff)
downloadnnfw-89b50c6bfaa9c54c6dde445894843cd433dea6a9.tar.gz
nnfw-89b50c6bfaa9c54c6dde445894843cd433dea6a9.tar.bz2
nnfw-89b50c6bfaa9c54c6dde445894843cd433dea6a9.zip
[res] Small network mocking the Epilogue (#6375)
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>
Diffstat (limited to 'res/TensorFlowTests')
-rw-r--r--res/TensorFlowTests/TF_SMALL_NET_0014/test.info2
-rw-r--r--res/TensorFlowTests/TF_SMALL_NET_0014/test.pbtxt106
2 files changed, 108 insertions, 0 deletions
diff --git a/res/TensorFlowTests/TF_SMALL_NET_0014/test.info b/res/TensorFlowTests/TF_SMALL_NET_0014/test.info
new file mode 100644
index 000000000..d3ea85a10
--- /dev/null
+++ b/res/TensorFlowTests/TF_SMALL_NET_0014/test.info
@@ -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
index 000000000..d4bc8e698
--- /dev/null
+++ b/res/TensorFlowTests/TF_SMALL_NET_0014/test.pbtxt
@@ -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 }
+ }
+}