From f26e48db4337bfbaf567ac21c72ff2d58d870040 Mon Sep 17 00:00:00 2001 From: Jeremy Johnson Date: Fri, 17 Jun 2022 13:07:12 +0100 Subject: Updating ARGMAX tests for v0.30.0 release Signed-off-by: Jeremy Johnson Change-Id: I5ad4da95fb9e88ab93634afcc3a4348771c711a5 --- .../test.json | 55 ++++++++++++++++++++++ 1 file changed, 55 insertions(+) create mode 100644 operators/tensor/argmax/argmax_ERRORIF_ArgmaxOutputShapeMismatch_15x19x19x17_i8_axis0/test.json (limited to 'operators/tensor/argmax/argmax_ERRORIF_ArgmaxOutputShapeMismatch_15x19x19x17_i8_axis0/test.json') diff --git a/operators/tensor/argmax/argmax_ERRORIF_ArgmaxOutputShapeMismatch_15x19x19x17_i8_axis0/test.json b/operators/tensor/argmax/argmax_ERRORIF_ArgmaxOutputShapeMismatch_15x19x19x17_i8_axis0/test.json new file mode 100644 index 000000000..f5565faf9 --- /dev/null +++ b/operators/tensor/argmax/argmax_ERRORIF_ArgmaxOutputShapeMismatch_15x19x19x17_i8_axis0/test.json @@ -0,0 +1,55 @@ +{ + version: { + _major: 0, + _minor: 30, + _patch: 0, + _draft: false + }, + blocks: [ + { + name: "main", + operators: [ + { + op: "ARGMAX", + attribute_type: "AxisAttribute", + attribute: { + axis: 0 + }, + inputs: [ + "input-0" + ], + outputs: [ + "result-0" + ] + } + ], + tensors: [ + { + name: "input-0", + shape: [ + 15, + 19, + 19, + 17 + ], + type: "INT8" + }, + { + name: "result-0", + shape: [ + 23, + 22, + 25 + ], + type: "INT32" + } + ], + inputs: [ + "input-0" + ], + outputs: [ + "result-0" + ] + } + ] +} -- cgit v1.2.1