summaryrefslogtreecommitdiff
path: root/res/TensorFlowTests/NET_0013/test.pbtxt
diff options
context:
space:
mode:
Diffstat (limited to 'res/TensorFlowTests/NET_0013/test.pbtxt')
-rw-r--r--res/TensorFlowTests/NET_0013/test.pbtxt72
1 files changed, 72 insertions, 0 deletions
diff --git a/res/TensorFlowTests/NET_0013/test.pbtxt b/res/TensorFlowTests/NET_0013/test.pbtxt
new file mode 100644
index 000000000..a9c6aff98
--- /dev/null
+++ b/res/TensorFlowTests/NET_0013/test.pbtxt
@@ -0,0 +1,72 @@
+# Network containing Shape node, which becomes Reshape's 'shape' input
+# Note that this is a work-around to test Shape
+#
+# (tensor)
+# Placeholder ---------- Reshape
+# /
+# Const --- Shape ---
+# (shape)
+node {
+ name: "Placeholder"
+ op: "Placeholder"
+ attr {
+ key: "dtype"
+ value { type: DT_FLOAT }
+ }
+ attr {
+ key: "shape"
+ value {
+ shape {
+ dim { size: 6 }
+ }
+ }
+ }
+}
+node {
+ name: "Const"
+ op: "Const"
+ attr {
+ key: "dtype"
+ value { type: DT_FLOAT }
+ }
+ attr {
+ key: "value"
+ value {
+ tensor {
+ dtype: DT_FLOAT
+ tensor_shape {
+ dim { size: 2 }
+ dim { size: 3 }
+ }
+ float_val: 1.0
+ }
+ }
+ }
+}
+node {
+ name: "Shape"
+ op: "Shape"
+ input: "Const"
+ attr {
+ key: "T"
+ value { type: DT_FLOAT }
+ }
+ attr {
+ key: "out_type"
+ value { type: DT_INT32 }
+ }
+}
+node {
+ name: "Reshape"
+ op: "Reshape"
+ input: "Placeholder"
+ input: "Shape"
+ attr {
+ key: "T"
+ value { type: DT_FLOAT }
+ }
+ attr {
+ key: "Tshape"
+ value { type: DT_INT32 }
+ }
+}