diff options
Diffstat (limited to 'res/TensorFlowLiteRecipes/ArgMax_U8_000/test.recipe')
-rw-r--r-- | res/TensorFlowLiteRecipes/ArgMax_U8_000/test.recipe | 31 |
1 files changed, 31 insertions, 0 deletions
diff --git a/res/TensorFlowLiteRecipes/ArgMax_U8_000/test.recipe b/res/TensorFlowLiteRecipes/ArgMax_U8_000/test.recipe new file mode 100644 index 000000000..12e2b250b --- /dev/null +++ b/res/TensorFlowLiteRecipes/ArgMax_U8_000/test.recipe @@ -0,0 +1,31 @@ +operand { + name: "ifm" + type: UINT8 + shape { dim: 4 } + quant { min: 0 max: 255 scale: 1.0 zero_point: 0 } +} +operand { + name: "ofm" + type: INT64 + shape { } +} +operand { + name: "argmax/dim" + type: INT32 + shape { } + filler { + tag: "explicit" + arg: "0" + } +} +operation { + type: "ArgMax" + argmax_options { + output_type: INT64 + } + input: "ifm" + input: "argmax/dim" + output: "ofm" +} +input: "ifm" +output: "ofm" |