summaryrefslogtreecommitdiff
path: root/res/TensorFlowTests
diff options
context:
space:
mode:
author박세희/On-Device Lab(SR)/Principal Engineer/삼성전자 <saehie.park@samsung.com>2019-09-05 14:25:07 +0900
committerGitHub Enterprise <noreply-CODE@samsung.com>2019-09-05 14:25:07 +0900
commit15910d5f02c34ee9aa93ebdd785453a7eb67596f (patch)
treef906b40e53e4f2b5a628ca36dd1736ef4371fb68 /res/TensorFlowTests
parent8c6940ae5029c0739d0ce0f30078f97b1a13a921 (diff)
downloadnnfw-15910d5f02c34ee9aa93ebdd785453a7eb67596f.tar.gz
nnfw-15910d5f02c34ee9aa93ebdd785453a7eb67596f.tar.bz2
nnfw-15910d5f02c34ee9aa93ebdd785453a7eb67596f.zip
[res] TF test for rank 4 StopGradient (#7191)
This will add TensorFlow test material for rank 4 StopGradient operator Signed-off-by: SaeHie Park <saehie.park@samsung.com>
Diffstat (limited to 'res/TensorFlowTests')
-rw-r--r--res/TensorFlowTests/UNIT_StopGradient_001/test.info2
-rw-r--r--res/TensorFlowTests/UNIT_StopGradient_001/test.pbtxt40
2 files changed, 42 insertions, 0 deletions
diff --git a/res/TensorFlowTests/UNIT_StopGradient_001/test.info b/res/TensorFlowTests/UNIT_StopGradient_001/test.info
new file mode 100644
index 000000000..8f87a218f
--- /dev/null
+++ b/res/TensorFlowTests/UNIT_StopGradient_001/test.info
@@ -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
index 000000000..95e7c358c
--- /dev/null
+++ b/res/TensorFlowTests/UNIT_StopGradient_001/test.pbtxt
@@ -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
+ }
+ }
+}