diff options
-rw-r--r-- | res/TensorFlowTests/Squeeze_000/test.info | 2 | ||||
-rw-r--r-- | res/TensorFlowTests/Squeeze_000/test.pbtxt | 28 |
2 files changed, 30 insertions, 0 deletions
diff --git a/res/TensorFlowTests/Squeeze_000/test.info b/res/TensorFlowTests/Squeeze_000/test.info new file mode 100644 index 000000000..fd07403f2 --- /dev/null +++ b/res/TensorFlowTests/Squeeze_000/test.info @@ -0,0 +1,2 @@ +input, Placeholder:0, TF_FLOAT, [1, 4, 4, 3] +output, Squeeze:0, TF_FLOAT, [4, 4, 3] diff --git a/res/TensorFlowTests/Squeeze_000/test.pbtxt b/res/TensorFlowTests/Squeeze_000/test.pbtxt new file mode 100644 index 000000000..5ad75fca6 --- /dev/null +++ b/res/TensorFlowTests/Squeeze_000/test.pbtxt @@ -0,0 +1,28 @@ +node { + name: "Placeholder" + op: "Placeholder" + attr { + key: "dtype" + value { type: DT_FLOAT } + } + attr { + key: "shape" + value { + shape { + dim { size: 1 } + dim { size: 4 } + dim { size: 4 } + dim { size: 3 } + } + } + } +} +node { + name: "Squeeze" + op: "Squeeze" + input: "Placeholder" + attr { + key: "T" + value { type: DT_FLOAT } + } +} |