summaryrefslogtreecommitdiff
path: root/res/TensorFlowTests/NET_0020/test.pbtxt
diff options
context:
space:
mode:
Diffstat (limited to 'res/TensorFlowTests/NET_0020/test.pbtxt')
-rw-r--r--res/TensorFlowTests/NET_0020/test.pbtxt112
1 files changed, 112 insertions, 0 deletions
diff --git a/res/TensorFlowTests/NET_0020/test.pbtxt b/res/TensorFlowTests/NET_0020/test.pbtxt
new file mode 100644
index 000000000..efd18d132
--- /dev/null
+++ b/res/TensorFlowTests/NET_0020/test.pbtxt
@@ -0,0 +1,112 @@
+# The Epilogue, or endmost part of inception v3 comprised of Squeeze,
+# Reshape, Shape and Softmax
+#
+# Only difference from original is input shape:
+# - original has unknown batch and 1001 channels [?, 1, 1, 1001]
+# - this test has 2 batches and 3 channels [2, 1, 1, 3]
+
+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: "softmax"
+ op: "Softmax"
+ 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: "softmax"
+ input: "shape"
+ attr {
+ key: "T"
+ value { type: DT_FLOAT }
+ }
+ attr {
+ key: "Tshape"
+ value { type: DT_INT32 }
+ }
+}