diff options
author | 남궁석/On-Device Lab(SR)/Engineer/삼성전자 <sk.namkoong@samsung.com> | 2019-08-30 09:04:20 +0900 |
---|---|---|
committer | 박종현/On-Device Lab(SR)/Staff Engineer/삼성전자 <jh1302.park@samsung.com> | 2019-08-30 09:04:20 +0900 |
commit | a41afe61de727cbcb0142411be9589e56d3b5d14 (patch) | |
tree | cf69e97b2f2a72865a3e5189bd5d0034dd7f8405 /res | |
parent | cde49e06ba2a0c63f4ace393d09be226d3bc0e3e (diff) | |
download | nnfw-a41afe61de727cbcb0142411be9589e56d3b5d14.tar.gz nnfw-a41afe61de727cbcb0142411be9589e56d3b5d14.tar.bz2 nnfw-a41afe61de727cbcb0142411be9589e56d3b5d14.zip |
[res] Introduce UNIT_Squeeze_003 test (#7039)
This test is for checking that `Squeeze` operation handle multiple negative
squeeze_dims or not
Signed-off-by: Seok NamKoong <sk.namkoong@samsung.com>
Diffstat (limited to 'res')
-rw-r--r-- | res/TensorFlowTests/UNIT_Squeeze_003/test.info | 2 | ||||
-rw-r--r-- | res/TensorFlowTests/UNIT_Squeeze_003/test.pbtxt | 35 |
2 files changed, 37 insertions, 0 deletions
diff --git a/res/TensorFlowTests/UNIT_Squeeze_003/test.info b/res/TensorFlowTests/UNIT_Squeeze_003/test.info new file mode 100644 index 000000000..7cc93f354 --- /dev/null +++ b/res/TensorFlowTests/UNIT_Squeeze_003/test.info @@ -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 index 000000000..1c760c87f --- /dev/null +++ b/res/TensorFlowTests/UNIT_Squeeze_003/test.pbtxt @@ -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 } + } + } +} |