summaryrefslogtreecommitdiff
path: root/res/TensorFlowLiteRecipes/ArgMax_003/test.recipe
diff options
context:
space:
mode:
Diffstat (limited to 'res/TensorFlowLiteRecipes/ArgMax_003/test.recipe')
-rw-r--r--res/TensorFlowLiteRecipes/ArgMax_003/test.recipe30
1 files changed, 30 insertions, 0 deletions
diff --git a/res/TensorFlowLiteRecipes/ArgMax_003/test.recipe b/res/TensorFlowLiteRecipes/ArgMax_003/test.recipe
new file mode 100644
index 000000000..cb34e3824
--- /dev/null
+++ b/res/TensorFlowLiteRecipes/ArgMax_003/test.recipe
@@ -0,0 +1,30 @@
+operand {
+ name: "ifm"
+ type: FLOAT32
+ shape { dim: 4 dim: 5 dim: 6 }
+}
+operand {
+ name: "ofm"
+ type: INT64
+ shape { dim: 4 dim: 6 }
+}
+operand {
+ name: "argmax/dim"
+ type: INT32
+ shape { }
+ filler {
+ tag: "explicit"
+ arg: "1"
+ }
+}
+operation {
+ type: "ArgMax"
+ argmax_options {
+ output_type: INT64
+ }
+ input: "ifm"
+ input: "argmax/dim"
+ output: "ofm"
+}
+input: "ifm"
+output: "ofm"