diff options
author | 박세희/On-Device Lab(SR)/Principal Engineer/삼성전자 <saehie.park@samsung.com> | 2019-09-05 14:25:07 +0900 |
---|---|---|
committer | GitHub Enterprise <noreply-CODE@samsung.com> | 2019-09-05 14:25:07 +0900 |
commit | 15910d5f02c34ee9aa93ebdd785453a7eb67596f (patch) | |
tree | f906b40e53e4f2b5a628ca36dd1736ef4371fb68 /res/TensorFlowTests | |
parent | 8c6940ae5029c0739d0ce0f30078f97b1a13a921 (diff) | |
download | nnfw-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.info | 2 | ||||
-rw-r--r-- | res/TensorFlowTests/UNIT_StopGradient_001/test.pbtxt | 40 |
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 + } + } +} |