diff options
Diffstat (limited to 'res/TensorFlowTests/NET_0020/test.pbtxt')
-rw-r--r-- | res/TensorFlowTests/NET_0020/test.pbtxt | 112 |
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 } + } +} |