diff options
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 |
commit | 89b50c6bfaa9c54c6dde445894843cd433dea6a9 (patch) | |
tree | 76ebc173b4e4fe5e793b287bdc4fb1f7f9264361 /res/TensorFlowTests | |
parent | e8019fc78435b2f4e88d33f1f31522f8271c545f (diff) | |
download | nnfw-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.info | 2 | ||||
-rw-r--r-- | res/TensorFlowTests/TF_SMALL_NET_0014/test.pbtxt | 106 |
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 } + } +} |