diff options
Diffstat (limited to 'operators/tensor/argmax/argmax_ERRORIF_ArgmaxOutputShapeMismatch_10x11x2x5_i16_axis0/test.json')
| -rw-r--r-- | operators/tensor/argmax/argmax_ERRORIF_ArgmaxOutputShapeMismatch_10x11x2x5_i16_axis0/test.json | 73 |
1 files changed, 39 insertions, 34 deletions
diff --git a/operators/tensor/argmax/argmax_ERRORIF_ArgmaxOutputShapeMismatch_10x11x2x5_i16_axis0/test.json b/operators/tensor/argmax/argmax_ERRORIF_ArgmaxOutputShapeMismatch_10x11x2x5_i16_axis0/test.json index a3523338b..14853247e 100644 --- a/operators/tensor/argmax/argmax_ERRORIF_ArgmaxOutputShapeMismatch_10x11x2x5_i16_axis0/test.json +++ b/operators/tensor/argmax/argmax_ERRORIF_ArgmaxOutputShapeMismatch_10x11x2x5_i16_axis0/test.json @@ -5,16 +5,48 @@ _patch: 0, _draft: true }, - blocks: [ + regions: [ { name: "main", - operators: [ + blocks: [ { - op: "ARGMAX", - attribute_type: "AxisAttribute", - attribute: { - axis: 0 - }, + name: "main", + operators: [ + { + op: "ARGMAX", + attribute_type: "AxisAttribute", + attribute: { + axis: 0 + }, + inputs: [ + "input-0" + ], + outputs: [ + "result-0" + ] + } + ], + tensors: [ + { + name: "input-0", + shape: [ + 10, + 11, + 2, + 5 + ], + type: "INT16" + }, + { + name: "result-0", + shape: [ + 18, + 11, + 6 + ], + type: "INT32" + } + ], inputs: [ "input-0" ], @@ -22,33 +54,6 @@ "result-0" ] } - ], - tensors: [ - { - name: "input-0", - shape: [ - 10, - 11, - 2, - 5 - ], - type: "INT16" - }, - { - name: "result-0", - shape: [ - 18, - 11, - 6 - ], - type: "INT32" - } - ], - inputs: [ - "input-0" - ], - outputs: [ - "result-0" ] } ] |