[res] TF_SMALL_NET_0007 with Conv2D+Mul (#4200)
author박세희/On-Device Lab(SR)/Principal Engineer/삼성전자 <saehie.park@samsung.com>
Thu, 11 Jul 2019 06:56:39 +0000 (15:56 +0900)
committerGitHub Enterprise <noreply-CODE@samsung.com>
Thu, 11 Jul 2019 06:56:39 +0000 (15:56 +0900)
This will introduce TF_SMALL_NET_0007 having Conv2D and Mul node
This is to check FuseBinaryIntoPreceding transformation that fuses Mul into Conv2D kernel and be removed

Signed-off-by: SaeHie Park <saehie.park@samsung.com>
res/TensorFlowTests/TF_SMALL_NET_0007/test.info [new file with mode: 0644]
res/TensorFlowTests/TF_SMALL_NET_0007/test.pbtxt [new file with mode: 0644]

diff --git a/res/TensorFlowTests/TF_SMALL_NET_0007/test.info b/res/TensorFlowTests/TF_SMALL_NET_0007/test.info
new file mode 100644 (file)
index 0000000..59c3f61
--- /dev/null
@@ -0,0 +1,2 @@
+input, Placeholder:0, TF_FLOAT, [1, 3, 3, 5]
+output, Mul:0, TF_FLOAT, [1, 3, 3, 2]
diff --git a/res/TensorFlowTests/TF_SMALL_NET_0007/test.pbtxt b/res/TensorFlowTests/TF_SMALL_NET_0007/test.pbtxt
new file mode 100644 (file)
index 0000000..10f9f35
--- /dev/null
@@ -0,0 +1,151 @@
+# A simple network that has "Conv2D" + "Mul"
+node {
+  name: "Placeholder"
+  op: "Placeholder"
+  attr {
+    key: "dtype"
+    value {
+      type: DT_FLOAT
+    }
+  }
+  attr {
+    key: "shape"
+    value {
+      shape {
+        dim {
+          size: 1
+        }
+        dim {
+          size: 3
+        }
+        dim {
+          size: 3
+        }
+        dim {
+          size: 5
+        }
+      }
+    }
+  }
+}
+node {
+  name: "weights"
+  op: "Const"
+  attr {
+    key: "dtype"
+    value {
+      type: DT_FLOAT
+    }
+  }
+  attr {
+    key: "value"
+    value {
+      tensor {
+        dtype: DT_FLOAT
+        tensor_shape {
+          dim {
+            size: 2
+          }
+          dim {
+            size: 2
+          }
+          dim {
+            size: 5
+          }
+          dim {
+            size: 2
+          }
+        }
+        float_val: 1.100000023841858
+      }
+    }
+  }
+}
+node {
+  name: "Conv2D"
+  op: "Conv2D"
+  input: "Placeholder"
+  input: "weights"
+  attr {
+    key: "T"
+    value {
+      type: DT_FLOAT
+    }
+  }
+  attr {
+    key: "data_format"
+    value {
+      s: "NHWC"
+    }
+  }
+  attr {
+    key: "dilations"
+    value {
+      list {
+        i: 1
+        i: 1
+        i: 1
+        i: 1
+      }
+    }
+  }
+  attr {
+    key: "padding"
+    value {
+      s: "SAME"
+    }
+  }
+  attr {
+    key: "strides"
+    value {
+      list {
+        i: 1
+        i: 1
+        i: 1
+        i: 1
+      }
+    }
+  }
+  attr {
+    key: "use_cudnn_on_gpu"
+    value {
+      b: true
+    }
+  }
+}
+node {
+  name: "mulparam"
+  op: "Const"
+  attr {
+    key: "dtype"
+    value {
+      type: DT_FLOAT
+    }
+  }
+  attr {
+    key: "value"
+    value {
+      tensor {
+        dtype: DT_FLOAT
+        tensor_shape {
+          dim {
+            size: 2
+          }
+        }
+        float_val: 2.0
+      }
+    }
+  }
+}
+node {
+  name: "Mul"
+  op: "Mul"
+  input: "Conv2D"
+  input: "mulparam"
+  attr {
+    key: "T"
+    value {
+      type: DT_FLOAT
+    }
+  }
+}