[res] TF test for rank 4 StopGradient (#7191)
author박세희/On-Device Lab(SR)/Principal Engineer/삼성전자 <saehie.park@samsung.com>
Thu, 5 Sep 2019 05:25:07 +0000 (14:25 +0900)
committerGitHub Enterprise <noreply-CODE@samsung.com>
Thu, 5 Sep 2019 05:25:07 +0000 (14:25 +0900)
This will add TensorFlow test material for rank 4 StopGradient operator

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

diff --git a/res/TensorFlowTests/UNIT_StopGradient_001/test.info b/res/TensorFlowTests/UNIT_StopGradient_001/test.info
new file mode 100644 (file)
index 0000000..8f87a21
--- /dev/null
@@ -0,0 +1,2 @@
+input,  Placeholder:0,   TF_FLOAT,   [1, 3, 3, 2]
+output, StopGradient:0,  TF_FLOAT,   [1, 3, 3, 2]
diff --git a/res/TensorFlowTests/UNIT_StopGradient_001/test.pbtxt b/res/TensorFlowTests/UNIT_StopGradient_001/test.pbtxt
new file mode 100644 (file)
index 0000000..95e7c35
--- /dev/null
@@ -0,0 +1,40 @@
+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: 2
+        }
+      }
+    }
+  }
+}
+node {
+  name: "StopGradient"
+  op: "StopGradient"
+  input: "Placeholder"
+  attr {
+    key: "T"
+    value {
+      type: DT_FLOAT
+    }
+  }
+}