From 2de8fa3ebe19745e9f2828f4b8d618d9fcd7dc0c Mon Sep 17 00:00:00 2001 From: =?utf8?q?=EB=B0=95=EC=B2=9C=EA=B5=90/On-Device=20Lab=28SR=29/Enginee?= =?utf8?q?r/=EC=82=BC=EC=84=B1=EC=A0=84=EC=9E=90?= Date: Mon, 5 Aug 2019 19:20:59 +0900 Subject: [PATCH] [res] Add selective Squeeze test (#6230) Add Squeeze test that has selective squeeze dimension Signed-off-by: Cheongyo Bahk --- res/TensorFlowTests/Squeeze_001/test.info | 2 ++ res/TensorFlowTests/Squeeze_001/test.pbtxt | 35 ++++++++++++++++++++++++++++++ 2 files changed, 37 insertions(+) create mode 100644 res/TensorFlowTests/Squeeze_001/test.info create mode 100644 res/TensorFlowTests/Squeeze_001/test.pbtxt diff --git a/res/TensorFlowTests/Squeeze_001/test.info b/res/TensorFlowTests/Squeeze_001/test.info new file mode 100644 index 0000000..f71e97f --- /dev/null +++ b/res/TensorFlowTests/Squeeze_001/test.info @@ -0,0 +1,2 @@ +input, Placeholder:0, TF_FLOAT, [2, 1, 3, 1] +output, Squeeze:0, TF_FLOAT, [2, 3, 1] diff --git a/res/TensorFlowTests/Squeeze_001/test.pbtxt b/res/TensorFlowTests/Squeeze_001/test.pbtxt new file mode 100644 index 0000000..641300c --- /dev/null +++ b/res/TensorFlowTests/Squeeze_001/test.pbtxt @@ -0,0 +1,35 @@ +# Network with Squeeze that has selective squeeze dimension +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 } + } + } +} -- 2.7.4