summaryrefslogtreecommitdiff
path: root/res/TensorFlowTests/UNIT_Softmax_003/test.pbtxt
diff options
context:
space:
mode:
Diffstat (limited to 'res/TensorFlowTests/UNIT_Softmax_003/test.pbtxt')
-rw-r--r--res/TensorFlowTests/UNIT_Softmax_003/test.pbtxt41
1 files changed, 41 insertions, 0 deletions
diff --git a/res/TensorFlowTests/UNIT_Softmax_003/test.pbtxt b/res/TensorFlowTests/UNIT_Softmax_003/test.pbtxt
new file mode 100644
index 000000000..c4b5ca27f
--- /dev/null
+++ b/res/TensorFlowTests/UNIT_Softmax_003/test.pbtxt
@@ -0,0 +1,41 @@
+# Softmax with rank-4 tensor
+node {
+ name: "Placeholder"
+ op: "Placeholder"
+ attr {
+ key: "dtype"
+ value {
+ type: DT_FLOAT
+ }
+ }
+ attr {
+ key: "shape"
+ value {
+ shape {
+ dim {
+ size: 2
+ }
+ dim {
+ size: 2
+ }
+ dim {
+ size: 3
+ }
+ dim {
+ size: 4
+ }
+ }
+ }
+ }
+}
+node {
+ name: "Softmax"
+ op: "Softmax"
+ input: "Placeholder"
+ attr {
+ key: "T"
+ value {
+ type: DT_FLOAT
+ }
+ }
+}