diff options
Diffstat (limited to 'res/TensorFlowTests/NET_0005/test.pbtxt')
-rw-r--r-- | res/TensorFlowTests/NET_0005/test.pbtxt | 120 |
1 files changed, 120 insertions, 0 deletions
diff --git a/res/TensorFlowTests/NET_0005/test.pbtxt b/res/TensorFlowTests/NET_0005/test.pbtxt new file mode 100644 index 000000000..16cc44005 --- /dev/null +++ b/res/TensorFlowTests/NET_0005/test.pbtxt @@ -0,0 +1,120 @@ +# Small Shape + Reshape network +# This test is quite similar with the last part of InceptionV3 +node { + name: "Placeholder" + op: "Placeholder" + attr { + key: "dtype" + value { + type: DT_FLOAT + } + } + attr { + key: "shape" + value { + shape { + dim { + size: 1 + } + dim { + size: 2 + } + dim { + size: 3 + } + dim { + size: 2 + } + } + } + } +} +node { + name: "Shape" + op: "Shape" + input: "Placeholder" + attr { + key: "T" + value { + type: DT_FLOAT + } + } + attr { + key: "out_type" + value { + type: DT_INT32 + } + } +} +node { + name: "Const" + op: "Const" + attr { + key: "dtype" + value { + type: DT_INT32 + } + } + attr { + key: "value" + value { + tensor { + dtype: DT_INT32 + tensor_shape { + dim { + size: 2 + } + } + int_val: 3 + int_val: 4 + } + } + } +} +node { + name: "Reshape1" + op: "Reshape" + input: "Placeholder" + input: "Const" + attr { + key: "T" + value { + type: DT_FLOAT + } + } + attr { + key: "Tshape" + value { + type: DT_INT32 + } + } +} +node { + name: "Relu" + op: "Relu" + input: "Reshape1" + attr { + key: "T" + value { + type: DT_FLOAT + } + } +} +node { + name: "Reshape2" + op: "Reshape" + input: "Relu" + input: "Shape" + attr { + key: "T" + value { + type: DT_FLOAT + } + } + attr { + key: "Tshape" + value { + type: DT_INT32 + } + } +} |