diff options
Diffstat (limited to 'operators/tensor/argmax')
44 files changed, 1626 insertions, 1406 deletions
diff --git a/operators/tensor/argmax/argmax_14x27x3x22_bf16_axis0/test.json b/operators/tensor/argmax/argmax_14x27x3x22_bf16_axis0/test.json index cecbc8cd0..ef22a376c 100644 --- a/operators/tensor/argmax/argmax_14x27x3x22_bf16_axis0/test.json +++ b/operators/tensor/argmax/argmax_14x27x3x22_bf16_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: [ + 14, + 27, + 3, + 22 + ], + type: "BF16" + }, + { + name: "result-0", + shape: [ + 27, + 3, + 22 + ], + type: "INT32" + } + ], inputs: [ "input-0" ], @@ -22,33 +54,6 @@ "result-0" ] } - ], - tensors: [ - { - name: "input-0", - shape: [ - 14, - 27, - 3, - 22 - ], - type: "BF16" - }, - { - name: "result-0", - shape: [ - 27, - 3, - 22 - ], - type: "INT32" - } - ], - inputs: [ - "input-0" - ], - outputs: [ - "result-0" ] } ] diff --git a/operators/tensor/argmax/argmax_14x27x3x22_bf16_axis2/test.json b/operators/tensor/argmax/argmax_14x27x3x22_bf16_axis2/test.json index 572b9ed06..1967d3e6a 100644 --- a/operators/tensor/argmax/argmax_14x27x3x22_bf16_axis2/test.json +++ b/operators/tensor/argmax/argmax_14x27x3x22_bf16_axis2/test.json @@ -5,16 +5,48 @@ _patch: 0, _draft: true }, - blocks: [ + regions: [ { name: "main", - operators: [ + blocks: [ { - op: "ARGMAX", - attribute_type: "AxisAttribute", - attribute: { - axis: 2 - }, + name: "main", + operators: [ + { + op: "ARGMAX", + attribute_type: "AxisAttribute", + attribute: { + axis: 2 + }, + inputs: [ + "input-0" + ], + outputs: [ + "result-0" + ] + } + ], + tensors: [ + { + name: "input-0", + shape: [ + 14, + 27, + 3, + 22 + ], + type: "BF16" + }, + { + name: "result-0", + shape: [ + 14, + 27, + 22 + ], + type: "INT32" + } + ], inputs: [ "input-0" ], @@ -22,33 +54,6 @@ "result-0" ] } - ], - tensors: [ - { - name: "input-0", - shape: [ - 14, - 27, - 3, - 22 - ], - type: "BF16" - }, - { - name: "result-0", - shape: [ - 14, - 27, - 22 - ], - type: "INT32" - } - ], - inputs: [ - "input-0" - ], - outputs: [ - "result-0" ] } ] diff --git a/operators/tensor/argmax/argmax_14x27x3x22_bf16_axis3/test.json b/operators/tensor/argmax/argmax_14x27x3x22_bf16_axis3/test.json index adbd17613..8dd8f7646 100644 --- a/operators/tensor/argmax/argmax_14x27x3x22_bf16_axis3/test.json +++ b/operators/tensor/argmax/argmax_14x27x3x22_bf16_axis3/test.json @@ -5,16 +5,48 @@ _patch: 0, _draft: true }, - blocks: [ + regions: [ { name: "main", - operators: [ + blocks: [ { - op: "ARGMAX", - attribute_type: "AxisAttribute", - attribute: { - axis: 3 - }, + name: "main", + operators: [ + { + op: "ARGMAX", + attribute_type: "AxisAttribute", + attribute: { + axis: 3 + }, + inputs: [ + "input-0" + ], + outputs: [ + "result-0" + ] + } + ], + tensors: [ + { + name: "input-0", + shape: [ + 14, + 27, + 3, + 22 + ], + type: "BF16" + }, + { + name: "result-0", + shape: [ + 14, + 27, + 3 + ], + type: "INT32" + } + ], inputs: [ "input-0" ], @@ -22,33 +54,6 @@ "result-0" ] } - ], - tensors: [ - { - name: "input-0", - shape: [ - 14, - 27, - 3, - 22 - ], - type: "BF16" - }, - { - name: "result-0", - shape: [ - 14, - 27, - 3 - ], - type: "INT32" - } - ], - inputs: [ - "input-0" - ], - outputs: [ - "result-0" ] } ] diff --git a/operators/tensor/argmax/argmax_14x27x3x22_i16_axis0/test.json b/operators/tensor/argmax/argmax_14x27x3x22_i16_axis0/test.json index 08ff392ce..6cb04fb22 100644 --- a/operators/tensor/argmax/argmax_14x27x3x22_i16_axis0/test.json +++ b/operators/tensor/argmax/argmax_14x27x3x22_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: [ + 14, + 27, + 3, + 22 + ], + type: "INT16" + }, + { + name: "result-0", + shape: [ + 27, + 3, + 22 + ], + type: "INT32" + } + ], inputs: [ "input-0" ], @@ -22,33 +54,6 @@ "result-0" ] } - ], - tensors: [ - { - name: "input-0", - shape: [ - 14, - 27, - 3, - 22 - ], - type: "INT16" - }, - { - name: "result-0", - shape: [ - 27, - 3, - 22 - ], - type: "INT32" - } - ], - inputs: [ - "input-0" - ], - outputs: [ - "result-0" ] } ] diff --git a/operators/tensor/argmax/argmax_1x3x1x65535_i8_axis3/test.json b/operators/tensor/argmax/argmax_1x3x1x65535_i8_axis3/test.json index ddcdce863..933782111 100644 --- a/operators/tensor/argmax/argmax_1x3x1x65535_i8_axis3/test.json +++ b/operators/tensor/argmax/argmax_1x3x1x65535_i8_axis3/test.json @@ -5,16 +5,48 @@ _patch: 0, _draft: true }, - blocks: [ + regions: [ { name: "main", - operators: [ + blocks: [ { - op: "ARGMAX", - attribute_type: "AxisAttribute", - attribute: { - axis: 3 - }, + name: "main", + operators: [ + { + op: "ARGMAX", + attribute_type: "AxisAttribute", + attribute: { + axis: 3 + }, + inputs: [ + "input-0" + ], + outputs: [ + "result-0" + ] + } + ], + tensors: [ + { + name: "input-0", + shape: [ + 1, + 3, + 1, + 65535 + ], + type: "INT8" + }, + { + name: "result-0", + shape: [ + 1, + 3, + 1 + ], + type: "INT32" + } + ], inputs: [ "input-0" ], @@ -22,33 +54,6 @@ "result-0" ] } - ], - tensors: [ - { - name: "input-0", - shape: [ - 1, - 3, - 1, - 65535 - ], - type: "INT8" - }, - { - name: "result-0", - shape: [ - 1, - 3, - 1 - ], - type: "INT32" - } - ], - inputs: [ - "input-0" - ], - outputs: [ - "result-0" ] } ] diff --git a/operators/tensor/argmax/argmax_1x3x65535x1_f32_axis1/test.json b/operators/tensor/argmax/argmax_1x3x65535x1_f32_axis1/test.json index 5d13e7111..24df80de8 100644 --- a/operators/tensor/argmax/argmax_1x3x65535x1_f32_axis1/test.json +++ b/operators/tensor/argmax/argmax_1x3x65535x1_f32_axis1/test.json @@ -5,16 +5,48 @@ _patch: 0, _draft: true }, - blocks: [ + regions: [ { name: "main", - operators: [ + blocks: [ { - op: "ARGMAX", - attribute_type: "AxisAttribute", - attribute: { - axis: 1 - }, + name: "main", + operators: [ + { + op: "ARGMAX", + attribute_type: "AxisAttribute", + attribute: { + axis: 1 + }, + inputs: [ + "input-0" + ], + outputs: [ + "result-0" + ] + } + ], + tensors: [ + { + name: "input-0", + shape: [ + 1, + 3, + 65535, + 1 + ], + type: "FP32" + }, + { + name: "result-0", + shape: [ + 1, + 65535, + 1 + ], + type: "INT32" + } + ], inputs: [ "input-0" ], @@ -22,33 +54,6 @@ "result-0" ] } - ], - tensors: [ - { - name: "input-0", - shape: [ - 1, - 3, - 65535, - 1 - ], - type: "FP32" - }, - { - name: "result-0", - shape: [ - 1, - 65535, - 1 - ], - type: "INT32" - } - ], - inputs: [ - "input-0" - ], - outputs: [ - "result-0" ] } ] diff --git a/operators/tensor/argmax/argmax_1x65537x1x2_f32_axis0/test.json b/operators/tensor/argmax/argmax_1x65537x1x2_f32_axis0/test.json index 98124c211..340adfffb 100644 --- a/operators/tensor/argmax/argmax_1x65537x1x2_f32_axis0/test.json +++ b/operators/tensor/argmax/argmax_1x65537x1x2_f32_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: [ + 1, + 65537, + 1, + 2 + ], + type: "FP32" + }, + { + name: "result-0", + shape: [ + 65537, + 1, + 2 + ], + type: "INT32" + } + ], inputs: [ "input-0" ], @@ -22,33 +54,6 @@ "result-0" ] } - ], - tensors: [ - { - name: "input-0", - shape: [ - 1, - 65537, - 1, - 2 - ], - type: "FP32" - }, - { - name: "result-0", - shape: [ - 65537, - 1, - 2 - ], - type: "INT32" - } - ], - inputs: [ - "input-0" - ], - outputs: [ - "result-0" ] } ] diff --git a/operators/tensor/argmax/argmax_1x65538x1x2_i8_axis1/test.json b/operators/tensor/argmax/argmax_1x65538x1x2_i8_axis1/test.json index 990ba08a2..56b9ddc9d 100644 --- a/operators/tensor/argmax/argmax_1x65538x1x2_i8_axis1/test.json +++ b/operators/tensor/argmax/argmax_1x65538x1x2_i8_axis1/test.json @@ -5,16 +5,48 @@ _patch: 0, _draft: true }, - blocks: [ + regions: [ { name: "main", - operators: [ + blocks: [ { - op: "ARGMAX", - attribute_type: "AxisAttribute", - attribute: { - axis: 1 - }, + name: "main", + operators: [ + { + op: "ARGMAX", + attribute_type: "AxisAttribute", + attribute: { + axis: 1 + }, + inputs: [ + "input-0" + ], + outputs: [ + "result-0" + ] + } + ], + tensors: [ + { + name: "input-0", + shape: [ + 1, + 65538, + 1, + 2 + ], + type: "INT8" + }, + { + name: "result-0", + shape: [ + 1, + 1, + 2 + ], + type: "INT32" + } + ], inputs: [ "input-0" ], @@ -22,33 +54,6 @@ "result-0" ] } - ], - tensors: [ - { - name: "input-0", - shape: [ - 1, - 65538, - 1, - 2 - ], - type: "INT8" - }, - { - name: "result-0", - shape: [ - 1, - 1, - 2 - ], - type: "INT32" - } - ], - inputs: [ - "input-0" - ], - outputs: [ - "result-0" ] } ] diff --git a/operators/tensor/argmax/argmax_34_f16_axis0/test.json b/operators/tensor/argmax/argmax_34_f16_axis0/test.json index c5e550099..6d2384a08 100644 --- a/operators/tensor/argmax/argmax_34_f16_axis0/test.json +++ b/operators/tensor/argmax/argmax_34_f16_axis0/test.json @@ -5,16 +5,43 @@ _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: [ + 34 + ], + type: "FP16" + }, + { + name: "result-0", + shape: [ + + ], + type: "INT32" + } + ], inputs: [ "input-0" ], @@ -22,28 +49,6 @@ "result-0" ] } - ], - tensors: [ - { - name: "input-0", - shape: [ - 34 - ], - type: "FP16" - }, - { - name: "result-0", - shape: [ - - ], - type: "INT32" - } - ], - inputs: [ - "input-0" - ], - outputs: [ - "result-0" ] } ] diff --git a/operators/tensor/argmax/argmax_34_i8_axis0/test.json b/operators/tensor/argmax/argmax_34_i8_axis0/test.json index 1a0aceb45..16e404cce 100644 --- a/operators/tensor/argmax/argmax_34_i8_axis0/test.json +++ b/operators/tensor/argmax/argmax_34_i8_axis0/test.json @@ -5,16 +5,43 @@ _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: [ + 34 + ], + type: "INT8" + }, + { + name: "result-0", + shape: [ + + ], + type: "INT32" + } + ], inputs: [ "input-0" ], @@ -22,28 +49,6 @@ "result-0" ] } - ], - tensors: [ - { - name: "input-0", - shape: [ - 34 - ], - type: "INT8" - }, - { - name: "result-0", - shape: [ - - ], - type: "INT32" - } - ], - inputs: [ - "input-0" - ], - outputs: [ - "result-0" ] } ] diff --git a/operators/tensor/argmax/argmax_34x54x38_i8_axis1/test.json b/operators/tensor/argmax/argmax_34x54x38_i8_axis1/test.json index 8a01dc859..f167f9431 100644 --- a/operators/tensor/argmax/argmax_34x54x38_i8_axis1/test.json +++ b/operators/tensor/argmax/argmax_34x54x38_i8_axis1/test.json @@ -5,16 +5,46 @@ _patch: 0, _draft: true }, - blocks: [ + regions: [ { name: "main", - operators: [ + blocks: [ { - op: "ARGMAX", - attribute_type: "AxisAttribute", - attribute: { - axis: 1 - }, + name: "main", + operators: [ + { + op: "ARGMAX", + attribute_type: "AxisAttribute", + attribute: { + axis: 1 + }, + inputs: [ + "input-0" + ], + outputs: [ + "result-0" + ] + } + ], + tensors: [ + { + name: "input-0", + shape: [ + 34, + 54, + 38 + ], + type: "INT8" + }, + { + name: "result-0", + shape: [ + 34, + 38 + ], + type: "INT32" + } + ], inputs: [ "input-0" ], @@ -22,31 +52,6 @@ "result-0" ] } - ], - tensors: [ - { - name: "input-0", - shape: [ - 34, - 54, - 38 - ], - type: "INT8" - }, - { - name: "result-0", - shape: [ - 34, - 38 - ], - type: "INT32" - } - ], - inputs: [ - "input-0" - ], - outputs: [ - "result-0" ] } ] diff --git a/operators/tensor/argmax/argmax_35x48x63_f16_axis1/test.json b/operators/tensor/argmax/argmax_35x48x63_f16_axis1/test.json index 002c3e632..b6dafe2fc 100644 --- a/operators/tensor/argmax/argmax_35x48x63_f16_axis1/test.json +++ b/operators/tensor/argmax/argmax_35x48x63_f16_axis1/test.json @@ -5,16 +5,46 @@ _patch: 0, _draft: true }, - blocks: [ + regions: [ { name: "main", - operators: [ + blocks: [ { - op: "ARGMAX", - attribute_type: "AxisAttribute", - attribute: { - axis: 1 - }, + name: "main", + operators: [ + { + op: "ARGMAX", + attribute_type: "AxisAttribute", + attribute: { + axis: 1 + }, + inputs: [ + "input-0" + ], + outputs: [ + "result-0" + ] + } + ], + tensors: [ + { + name: "input-0", + shape: [ + 35, + 48, + 63 + ], + type: "FP16" + }, + { + name: "result-0", + shape: [ + 35, + 63 + ], + type: "INT32" + } + ], inputs: [ "input-0" ], @@ -22,31 +52,6 @@ "result-0" ] } - ], - tensors: [ - { - name: "input-0", - shape: [ - 35, - 48, - 63 - ], - type: "FP16" - }, - { - name: "result-0", - shape: [ - 35, - 63 - ], - type: "INT32" - } - ], - inputs: [ - "input-0" - ], - outputs: [ - "result-0" ] } ] diff --git a/operators/tensor/argmax/argmax_35x48x63_i16_axis2/test.json b/operators/tensor/argmax/argmax_35x48x63_i16_axis2/test.json index e66a774a8..78fccffa5 100644 --- a/operators/tensor/argmax/argmax_35x48x63_i16_axis2/test.json +++ b/operators/tensor/argmax/argmax_35x48x63_i16_axis2/test.json @@ -5,16 +5,46 @@ _patch: 0, _draft: true }, - blocks: [ + regions: [ { name: "main", - operators: [ + blocks: [ { - op: "ARGMAX", - attribute_type: "AxisAttribute", - attribute: { - axis: 2 - }, + name: "main", + operators: [ + { + op: "ARGMAX", + attribute_type: "AxisAttribute", + attribute: { + axis: 2 + }, + inputs: [ + "input-0" + ], + outputs: [ + "result-0" + ] + } + ], + tensors: [ + { + name: "input-0", + shape: [ + 35, + 48, + 63 + ], + type: "INT16" + }, + { + name: "result-0", + shape: [ + 35, + 48 + ], + type: "INT32" + } + ], inputs: [ "input-0" ], @@ -22,31 +52,6 @@ "result-0" ] } - ], - tensors: [ - { - name: "input-0", - shape: [ - 35, - 48, - 63 - ], - type: "INT16" - }, - { - name: "result-0", - shape: [ - 35, - 48 - ], - type: "INT32" - } - ], - inputs: [ - "input-0" - ], - outputs: [ - "result-0" ] } ] diff --git a/operators/tensor/argmax/argmax_3x24x21x14_f16_axis0/test.json b/operators/tensor/argmax/argmax_3x24x21x14_f16_axis0/test.json index 40d92076c..a8ec2add0 100644 --- a/operators/tensor/argmax/argmax_3x24x21x14_f16_axis0/test.json +++ b/operators/tensor/argmax/argmax_3x24x21x14_f16_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: [ + 3, + 24, + 21, + 14 + ], + type: "FP16" + }, + { + name: "result-0", + shape: [ + 24, + 21, + 14 + ], + type: "INT32" + } + ], inputs: [ "input-0" ], @@ -22,33 +54,6 @@ "result-0" ] } - ], - tensors: [ - { - name: "input-0", - shape: [ - 3, - 24, - 21, - 14 - ], - type: "FP16" - }, - { - name: "result-0", - shape: [ - 24, - 21, - 14 - ], - type: "INT32" - } - ], - inputs: [ - "input-0" - ], - outputs: [ - "result-0" ] } ] diff --git a/operators/tensor/argmax/argmax_3x24x21x14_i8_axis1/test.json b/operators/tensor/argmax/argmax_3x24x21x14_i8_axis1/test.json index 622ce2f39..61a13b68b 100644 --- a/operators/tensor/argmax/argmax_3x24x21x14_i8_axis1/test.json +++ b/operators/tensor/argmax/argmax_3x24x21x14_i8_axis1/test.json @@ -5,16 +5,48 @@ _patch: 0, _draft: true }, - blocks: [ + regions: [ { name: "main", - operators: [ + blocks: [ { - op: "ARGMAX", - attribute_type: "AxisAttribute", - attribute: { - axis: 1 - }, + name: "main", + operators: [ + { + op: "ARGMAX", + attribute_type: "AxisAttribute", + attribute: { + axis: 1 + }, + inputs: [ + "input-0" + ], + outputs: [ + "result-0" + ] + } + ], + tensors: [ + { + name: "input-0", + shape: [ + 3, + 24, + 21, + 14 + ], + type: "INT8" + }, + { + name: "result-0", + shape: [ + 3, + 21, + 14 + ], + type: "INT32" + } + ], inputs: [ "input-0" ], @@ -22,33 +54,6 @@ "result-0" ] } - ], - tensors: [ - { - name: "input-0", - shape: [ - 3, - 24, - 21, - 14 - ], - type: "INT8" - }, - { - name: "result-0", - shape: [ - 3, - 21, - 14 - ], - type: "INT32" - } - ], - inputs: [ - "input-0" - ], - outputs: [ - "result-0" ] } ] diff --git a/operators/tensor/argmax/argmax_45x59_i16_axis0/test.json b/operators/tensor/argmax/argmax_45x59_i16_axis0/test.json index f311a04fd..0325ff50f 100644 --- a/operators/tensor/argmax/argmax_45x59_i16_axis0/test.json +++ b/operators/tensor/argmax/argmax_45x59_i16_axis0/test.json @@ -5,16 +5,44 @@ _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: [ + 45, + 59 + ], + type: "INT16" + }, + { + name: "result-0", + shape: [ + 59 + ], + type: "INT32" + } + ], inputs: [ "input-0" ], @@ -22,29 +50,6 @@ "result-0" ] } - ], - tensors: [ - { - name: "input-0", - shape: [ - 45, - 59 - ], - type: "INT16" - }, - { - name: "result-0", - shape: [ - 59 - ], - type: "INT32" - } - ], - inputs: [ - "input-0" - ], - outputs: [ - "result-0" ] } ] diff --git a/operators/tensor/argmax/argmax_46x45_f16_axis0/test.json b/operators/tensor/argmax/argmax_46x45_f16_axis0/test.json index 673942fc8..eea0cd44d 100644 --- a/operators/tensor/argmax/argmax_46x45_f16_axis0/test.json +++ b/operators/tensor/argmax/argmax_46x45_f16_axis0/test.json @@ -5,16 +5,44 @@ _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: [ + 46, + 45 + ], + type: "FP16" + }, + { + name: "result-0", + shape: [ + 45 + ], + type: "INT32" + } + ], inputs: [ "input-0" ], @@ -22,29 +50,6 @@ "result-0" ] } - ], - tensors: [ - { - name: "input-0", - shape: [ - 46, - 45 - ], - type: "FP16" - }, - { - name: "result-0", - shape: [ - 45 - ], - type: "INT32" - } - ], - inputs: [ - "input-0" - ], - outputs: [ - "result-0" ] } ] diff --git a/operators/tensor/argmax/argmax_52_f32_axis0/test.json b/operators/tensor/argmax/argmax_52_f32_axis0/test.json index baa92c600..d2feb1103 100644 --- a/operators/tensor/argmax/argmax_52_f32_axis0/test.json +++ b/operators/tensor/argmax/argmax_52_f32_axis0/test.json @@ -5,16 +5,43 @@ _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: [ + 52 + ], + type: "FP32" + }, + { + name: "result-0", + shape: [ + + ], + type: "INT32" + } + ], inputs: [ "input-0" ], @@ -22,28 +49,6 @@ "result-0" ] } - ], - tensors: [ - { - name: "input-0", - shape: [ - 52 - ], - type: "FP32" - }, - { - name: "result-0", - shape: [ - - ], - type: "INT32" - } - ], - inputs: [ - "input-0" - ], - outputs: [ - "result-0" ] } ] diff --git a/operators/tensor/argmax/argmax_52x46_i8_axis1/test.json b/operators/tensor/argmax/argmax_52x46_i8_axis1/test.json index cf43420a3..72a9e567d 100644 --- a/operators/tensor/argmax/argmax_52x46_i8_axis1/test.json +++ b/operators/tensor/argmax/argmax_52x46_i8_axis1/test.json @@ -5,16 +5,44 @@ _patch: 0, _draft: true }, - blocks: [ + regions: [ { name: "main", - operators: [ + blocks: [ { - op: "ARGMAX", - attribute_type: "AxisAttribute", - attribute: { - axis: 1 - }, + name: "main", + operators: [ + { + op: "ARGMAX", + attribute_type: "AxisAttribute", + attribute: { + axis: 1 + }, + inputs: [ + "input-0" + ], + outputs: [ + "result-0" + ] + } + ], + tensors: [ + { + name: "input-0", + shape: [ + 52, + 46 + ], + type: "INT8" + }, + { + name: "result-0", + shape: [ + 52 + ], + type: "INT32" + } + ], inputs: [ "input-0" ], @@ -22,29 +50,6 @@ "result-0" ] } - ], - tensors: [ - { - name: "input-0", - shape: [ - 52, - 46 - ], - type: "INT8" - }, - { - name: "result-0", - shape: [ - 52 - ], - type: "INT32" - } - ], - inputs: [ - "input-0" - ], - outputs: [ - "result-0" ] } ] diff --git a/operators/tensor/argmax/argmax_54x38_f32_axis0/test.json b/operators/tensor/argmax/argmax_54x38_f32_axis0/test.json index 05122a784..f972cb2bb 100644 --- a/operators/tensor/argmax/argmax_54x38_f32_axis0/test.json +++ b/operators/tensor/argmax/argmax_54x38_f32_axis0/test.json @@ -5,16 +5,44 @@ _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: [ + 54, + 38 + ], + type: "FP32" + }, + { + name: "result-0", + shape: [ + 38 + ], + type: "INT32" + } + ], inputs: [ "input-0" ], @@ -22,29 +50,6 @@ "result-0" ] } - ], - tensors: [ - { - name: "input-0", - shape: [ - 54, - 38 - ], - type: "FP32" - }, - { - name: "result-0", - shape: [ - 38 - ], - type: "INT32" - } - ], - inputs: [ - "input-0" - ], - outputs: [ - "result-0" ] } ] diff --git a/operators/tensor/argmax/argmax_55x56x54_bf16_axis0/test.json b/operators/tensor/argmax/argmax_55x56x54_bf16_axis0/test.json index 668cde7cc..04a447054 100644 --- a/operators/tensor/argmax/argmax_55x56x54_bf16_axis0/test.json +++ b/operators/tensor/argmax/argmax_55x56x54_bf16_axis0/test.json @@ -5,16 +5,46 @@ _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: [ + 55, + 56, + 54 + ], + type: "BF16" + }, + { + name: "result-0", + shape: [ + 56, + 54 + ], + type: "INT32" + } + ], inputs: [ "input-0" ], @@ -22,31 +52,6 @@ "result-0" ] } - ], - tensors: [ - { - name: "input-0", - shape: [ - 55, - 56, - 54 - ], - type: "BF16" - }, - { - name: "result-0", - shape: [ - 56, - 54 - ], - type: "INT32" - } - ], - inputs: [ - "input-0" - ], - outputs: [ - "result-0" ] } ] diff --git a/operators/tensor/argmax/argmax_56_bf16_axis0/test.json b/operators/tensor/argmax/argmax_56_bf16_axis0/test.json index f15cc6e45..cff960506 100644 --- a/operators/tensor/argmax/argmax_56_bf16_axis0/test.json +++ b/operators/tensor/argmax/argmax_56_bf16_axis0/test.json @@ -5,16 +5,43 @@ _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: [ + 56 + ], + type: "BF16" + }, + { + name: "result-0", + shape: [ + + ], + type: "INT32" + } + ], inputs: [ "input-0" ], @@ -22,28 +49,6 @@ "result-0" ] } - ], - tensors: [ - { - name: "input-0", - shape: [ - 56 - ], - type: "BF16" - }, - { - name: "result-0", - shape: [ - - ], - type: "INT32" - } - ], - inputs: [ - "input-0" - ], - outputs: [ - "result-0" ] } ] diff --git a/operators/tensor/argmax/argmax_56_i16_axis0/test.json b/operators/tensor/argmax/argmax_56_i16_axis0/test.json index a550a5b7b..52942d7e9 100644 --- a/operators/tensor/argmax/argmax_56_i16_axis0/test.json +++ b/operators/tensor/argmax/argmax_56_i16_axis0/test.json @@ -5,16 +5,43 @@ _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: [ + 56 + ], + type: "INT16" + }, + { + name: "result-0", + shape: [ + + ], + type: "INT32" + } + ], inputs: [ "input-0" ], @@ -22,28 +49,6 @@ "result-0" ] } - ], - tensors: [ - { - name: "input-0", - shape: [ - 56 - ], - type: "INT16" - }, - { - name: "result-0", - shape: [ - - ], - type: "INT32" - } - ], - inputs: [ - "input-0" - ], - outputs: [ - "result-0" ] } ] diff --git a/operators/tensor/argmax/argmax_57x48x36_f32_axis0/test.json b/operators/tensor/argmax/argmax_57x48x36_f32_axis0/test.json index 22c4ae52a..e346b00bc 100644 --- a/operators/tensor/argmax/argmax_57x48x36_f32_axis0/test.json +++ b/operators/tensor/argmax/argmax_57x48x36_f32_axis0/test.json @@ -5,16 +5,46 @@ _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: [ + 57, + 48, + 36 + ], + type: "FP32" + }, + { + name: "result-0", + shape: [ + 48, + 36 + ], + type: "INT32" + } + ], inputs: [ "input-0" ], @@ -22,31 +52,6 @@ "result-0" ] } - ], - tensors: [ - { - name: "input-0", - shape: [ - 57, - 48, - 36 - ], - type: "FP32" - }, - { - name: "result-0", - shape: [ - 48, - 36 - ], - type: "INT32" - } - ], - inputs: [ - "input-0" - ], - outputs: [ - "result-0" ] } ] diff --git a/operators/tensor/argmax/argmax_59x34_bf16_axis0/test.json b/operators/tensor/argmax/argmax_59x34_bf16_axis0/test.json index 7c5ea3b63..71ff6ee40 100644 --- a/operators/tensor/argmax/argmax_59x34_bf16_axis0/test.json +++ b/operators/tensor/argmax/argmax_59x34_bf16_axis0/test.json @@ -5,16 +5,44 @@ _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: [ + 59, + 34 + ], + type: "BF16" + }, + { + name: "result-0", + shape: [ + 34 + ], + type: "INT32" + } + ], inputs: [ "input-0" ], @@ -22,29 +50,6 @@ "result-0" ] } - ], - tensors: [ - { - name: "input-0", - shape: [ - 59, - 34 - ], - type: "BF16" - }, - { - name: "result-0", - shape: [ - 34 - ], - type: "INT32" - } - ], - inputs: [ - "input-0" - ], - outputs: [ - "result-0" ] } ] diff --git a/operators/tensor/argmax/argmax_7x3x17x31_f32_axis1/test.json b/operators/tensor/argmax/argmax_7x3x17x31_f32_axis1/test.json index 2ed0c26bc..f3b15e531 100644 --- a/operators/tensor/argmax/argmax_7x3x17x31_f32_axis1/test.json +++ b/operators/tensor/argmax/argmax_7x3x17x31_f32_axis1/test.json @@ -5,16 +5,48 @@ _patch: 0, _draft: true }, - blocks: [ + regions: [ { name: "main", - operators: [ + blocks: [ { - op: "ARGMAX", - attribute_type: "AxisAttribute", - attribute: { - axis: 1 - }, + name: "main", + operators: [ + { + op: "ARGMAX", + attribute_type: "AxisAttribute", + attribute: { + axis: 1 + }, + inputs: [ + "input-0" + ], + outputs: [ + "result-0" + ] + } + ], + tensors: [ + { + name: "input-0", + shape: [ + 7, + 3, + 17, + 31 + ], + type: "FP32" + }, + { + name: "result-0", + shape: [ + 7, + 17, + 31 + ], + type: "INT32" + } + ], inputs: [ "input-0" ], @@ -22,33 +54,6 @@ "result-0" ] } - ], - tensors: [ - { - name: "input-0", - shape: [ - 7, - 3, - 17, - 31 - ], - type: "FP32" - }, - { - name: "result-0", - shape: [ - 7, - 17, - 31 - ], - type: "INT32" - } - ], - inputs: [ - "input-0" - ], - outputs: [ - "result-0" ] } ] diff --git a/operators/tensor/argmax/argmax_ERRORIF_ArgmaxOutputRankMismatch_10_f16_axis0/test.json b/operators/tensor/argmax/argmax_ERRORIF_ArgmaxOutputRankMismatch_10_f16_axis0/test.json index 80cf42558..2f887df85 100644 --- a/operators/tensor/argmax/argmax_ERRORIF_ArgmaxOutputRankMismatch_10_f16_axis0/test.json +++ b/operators/tensor/argmax/argmax_ERRORIF_ArgmaxOutputRankMismatch_10_f16_axis0/test.json @@ -5,16 +5,43 @@ _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 + ], + type: "FP16" + }, + { + name: "result-0", + shape: [ + 1 + ], + type: "INT32" + } + ], inputs: [ "input-0" ], @@ -22,28 +49,6 @@ "result-0" ] } - ], - tensors: [ - { - name: "input-0", - shape: [ - 10 - ], - type: "FP16" - }, - { - name: "result-0", - shape: [ - 1 - ], - type: "INT32" - } - ], - inputs: [ - "input-0" - ], - outputs: [ - "result-0" ] } ] diff --git a/operators/tensor/argmax/argmax_ERRORIF_ArgmaxOutputRankMismatch_12x11_i8_axis0/test.json b/operators/tensor/argmax/argmax_ERRORIF_ArgmaxOutputRankMismatch_12x11_i8_axis0/test.json index 053d1a5fd..a4c128554 100644 --- a/operators/tensor/argmax/argmax_ERRORIF_ArgmaxOutputRankMismatch_12x11_i8_axis0/test.json +++ b/operators/tensor/argmax/argmax_ERRORIF_ArgmaxOutputRankMismatch_12x11_i8_axis0/test.json @@ -5,16 +5,45 @@ _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: [ + 12, + 11 + ], + type: "INT8" + }, + { + name: "result-0", + shape: [ + 11, + 1 + ], + type: "INT32" + } + ], inputs: [ "input-0" ], @@ -22,30 +51,6 @@ "result-0" ] } - ], - tensors: [ - { - name: "input-0", - shape: [ - 12, - 11 - ], - type: "INT8" - }, - { - name: "result-0", - shape: [ - 11, - 1 - ], - type: "INT32" - } - ], - inputs: [ - "input-0" - ], - outputs: [ - "result-0" ] } ] 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" ] } ] diff --git a/operators/tensor/argmax/argmax_ERRORIF_ArgmaxOutputShapeMismatch_13x9x2x12_bf16_axis0/test.json b/operators/tensor/argmax/argmax_ERRORIF_ArgmaxOutputShapeMismatch_13x9x2x12_bf16_axis0/test.json index 01dcb8a0a..d01c35da4 100644 --- a/operators/tensor/argmax/argmax_ERRORIF_ArgmaxOutputShapeMismatch_13x9x2x12_bf16_axis0/test.json +++ b/operators/tensor/argmax/argmax_ERRORIF_ArgmaxOutputShapeMismatch_13x9x2x12_bf16_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: [ + 13, + 9, + 2, + 12 + ], + type: "BF16" + }, + { + name: "result-0", + shape: [ + 18, + 7, + 19 + ], + type: "INT32" + } + ], inputs: [ "input-0" ], @@ -22,33 +54,6 @@ "result-0" ] } - ], - tensors: [ - { - name: "input-0", - shape: [ - 13, - 9, - 2, - 12 - ], - type: "BF16" - }, - { - name: "result-0", - shape: [ - 18, - 7, - 19 - ], - type: "INT32" - } - ], - inputs: [ - "input-0" - ], - outputs: [ - "result-0" ] } ] diff --git a/operators/tensor/argmax/argmax_ERRORIF_AxisLargerRank_11_f16_axis5/test.json b/operators/tensor/argmax/argmax_ERRORIF_AxisLargerRank_11_f16_axis5/test.json index 95349acbb..e0f689e25 100644 --- a/operators/tensor/argmax/argmax_ERRORIF_AxisLargerRank_11_f16_axis5/test.json +++ b/operators/tensor/argmax/argmax_ERRORIF_AxisLargerRank_11_f16_axis5/test.json @@ -5,16 +5,43 @@ _patch: 0, _draft: true }, - blocks: [ + regions: [ { name: "main", - operators: [ + blocks: [ { - op: "ARGMAX", - attribute_type: "AxisAttribute", - attribute: { - axis: 5 - }, + name: "main", + operators: [ + { + op: "ARGMAX", + attribute_type: "AxisAttribute", + attribute: { + axis: 5 + }, + inputs: [ + "input-0" + ], + outputs: [ + "result-0" + ] + } + ], + tensors: [ + { + name: "input-0", + shape: [ + 11 + ], + type: "FP16" + }, + { + name: "result-0", + shape: [ + 11 + ], + type: "INT32" + } + ], inputs: [ "input-0" ], @@ -22,28 +49,6 @@ "result-0" ] } - ], - tensors: [ - { - name: "input-0", - shape: [ - 11 - ], - type: "FP16" - }, - { - name: "result-0", - shape: [ - 11 - ], - type: "INT32" - } - ], - inputs: [ - "input-0" - ], - outputs: [ - "result-0" ] } ] diff --git a/operators/tensor/argmax/argmax_ERRORIF_AxisLargerRank_11x6x2_i16_axis12/test.json b/operators/tensor/argmax/argmax_ERRORIF_AxisLargerRank_11x6x2_i16_axis12/test.json index 8854be6a8..7c0cfcfdf 100644 --- a/operators/tensor/argmax/argmax_ERRORIF_AxisLargerRank_11x6x2_i16_axis12/test.json +++ b/operators/tensor/argmax/argmax_ERRORIF_AxisLargerRank_11x6x2_i16_axis12/test.json @@ -5,16 +5,47 @@ _patch: 0, _draft: true }, - blocks: [ + regions: [ { name: "main", - operators: [ + blocks: [ { - op: "ARGMAX", - attribute_type: "AxisAttribute", - attribute: { - axis: 12 - }, + name: "main", + operators: [ + { + op: "ARGMAX", + attribute_type: "AxisAttribute", + attribute: { + axis: 12 + }, + inputs: [ + "input-0" + ], + outputs: [ + "result-0" + ] + } + ], + tensors: [ + { + name: "input-0", + shape: [ + 11, + 6, + 2 + ], + type: "INT16" + }, + { + name: "result-0", + shape: [ + 11, + 6, + 2 + ], + type: "INT32" + } + ], inputs: [ "input-0" ], @@ -22,32 +53,6 @@ "result-0" ] } - ], - tensors: [ - { - name: "input-0", - shape: [ - 11, - 6, - 2 - ], - type: "INT16" - }, - { - name: "result-0", - shape: [ - 11, - 6, - 2 - ], - type: "INT32" - } - ], - inputs: [ - "input-0" - ], - outputs: [ - "result-0" ] } ] diff --git a/operators/tensor/argmax/argmax_ERRORIF_AxisSmallerZero_10_bf16_axis-3/test.json b/operators/tensor/argmax/argmax_ERRORIF_AxisSmallerZero_10_bf16_axis-3/test.json index cc7188a5c..10a3d914c 100644 --- a/operators/tensor/argmax/argmax_ERRORIF_AxisSmallerZero_10_bf16_axis-3/test.json +++ b/operators/tensor/argmax/argmax_ERRORIF_AxisSmallerZero_10_bf16_axis-3/test.json @@ -5,16 +5,43 @@ _patch: 0, _draft: true }, - blocks: [ + regions: [ { name: "main", - operators: [ + blocks: [ { - op: "ARGMAX", - attribute_type: "AxisAttribute", - attribute: { - axis: -3 - }, + name: "main", + operators: [ + { + op: "ARGMAX", + attribute_type: "AxisAttribute", + attribute: { + axis: -3 + }, + inputs: [ + "input-0" + ], + outputs: [ + "result-0" + ] + } + ], + tensors: [ + { + name: "input-0", + shape: [ + 10 + ], + type: "BF16" + }, + { + name: "result-0", + shape: [ + 10 + ], + type: "INT32" + } + ], inputs: [ "input-0" ], @@ -22,28 +49,6 @@ "result-0" ] } - ], - tensors: [ - { - name: "input-0", - shape: [ - 10 - ], - type: "BF16" - }, - { - name: "result-0", - shape: [ - 10 - ], - type: "INT32" - } - ], - inputs: [ - "input-0" - ], - outputs: [ - "result-0" ] } ] diff --git a/operators/tensor/argmax/argmax_ERRORIF_AxisSmallerZero_10_i16_axis-3/test.json b/operators/tensor/argmax/argmax_ERRORIF_AxisSmallerZero_10_i16_axis-3/test.json index 9f43780f8..513cf87ea 100644 --- a/operators/tensor/argmax/argmax_ERRORIF_AxisSmallerZero_10_i16_axis-3/test.json +++ b/operators/tensor/argmax/argmax_ERRORIF_AxisSmallerZero_10_i16_axis-3/test.json @@ -5,16 +5,43 @@ _patch: 0, _draft: true }, - blocks: [ + regions: [ { name: "main", - operators: [ + blocks: [ { - op: "ARGMAX", - attribute_type: "AxisAttribute", - attribute: { - axis: -3 - }, + name: "main", + operators: [ + { + op: "ARGMAX", + attribute_type: "AxisAttribute", + attribute: { + axis: -3 + }, + inputs: [ + "input-0" + ], + outputs: [ + "result-0" + ] + } + ], + tensors: [ + { + name: "input-0", + shape: [ + 10 + ], + type: "INT16" + }, + { + name: "result-0", + shape: [ + 10 + ], + type: "INT32" + } + ], inputs: [ "input-0" ], @@ -22,28 +49,6 @@ "result-0" ] } - ], - tensors: [ - { - name: "input-0", - shape: [ - 10 - ], - type: "INT16" - }, - { - name: "result-0", - shape: [ - 10 - ], - type: "INT32" - } - ], - inputs: [ - "input-0" - ], - outputs: [ - "result-0" ] } ] diff --git a/operators/tensor/argmax/argmax_ERRORIF_WrongInputList_10x10x2x12_f32_axis0/test.json b/operators/tensor/argmax/argmax_ERRORIF_WrongInputList_10x10x2x12_f32_axis0/test.json index ecc843ce7..2808ceedc 100644 --- a/operators/tensor/argmax/argmax_ERRORIF_WrongInputList_10x10x2x12_f32_axis0/test.json +++ b/operators/tensor/argmax/argmax_ERRORIF_WrongInputList_10x10x2x12_f32_axis0/test.json @@ -5,50 +5,55 @@ _patch: 0, _draft: true }, - blocks: [ + regions: [ { name: "main", - operators: [ + blocks: [ { - op: "ARGMAX", - attribute_type: "AxisAttribute", - attribute: { - axis: 0 - }, - inputs: [ + name: "main", + operators: [ + { + op: "ARGMAX", + attribute_type: "AxisAttribute", + attribute: { + axis: 0 + }, + inputs: [ + ], + outputs: [ + "result-0" + ] + } + ], + tensors: [ + { + name: "input-0", + shape: [ + 10, + 10, + 2, + 12 + ], + type: "FP32" + }, + { + name: "result-0", + shape: [ + 10, + 2, + 12 + ], + type: "INT32" + } + ], + inputs: [ + "input-0" ], outputs: [ "result-0" ] } - ], - tensors: [ - { - name: "input-0", - shape: [ - 10, - 10, - 2, - 12 - ], - type: "FP32" - }, - { - name: "result-0", - shape: [ - 10, - 2, - 12 - ], - type: "INT32" - } - ], - inputs: [ - "input-0" - ], - outputs: [ - "result-0" ] } ] diff --git a/operators/tensor/argmax/argmax_ERRORIF_WrongInputList_11x7x11x3_i16_axis0/test.json b/operators/tensor/argmax/argmax_ERRORIF_WrongInputList_11x7x11x3_i16_axis0/test.json index bed2509f8..64a4affee 100644 --- a/operators/tensor/argmax/argmax_ERRORIF_WrongInputList_11x7x11x3_i16_axis0/test.json +++ b/operators/tensor/argmax/argmax_ERRORIF_WrongInputList_11x7x11x3_i16_axis0/test.json @@ -5,51 +5,56 @@ _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", + "eiDummyInput" + ], + outputs: [ + "result-0" + ] + } + ], + tensors: [ + { + name: "input-0", + shape: [ + 11, + 7, + 11, + 3 + ], + type: "INT16" + }, + { + name: "result-0", + shape: [ + 7, + 11, + 3 + ], + type: "INT32" + } + ], inputs: [ - "input-0", - "eiDummyInput" + "input-0" ], outputs: [ "result-0" ] } - ], - tensors: [ - { - name: "input-0", - shape: [ - 11, - 7, - 11, - 3 - ], - type: "INT16" - }, - { - name: "result-0", - shape: [ - 7, - 11, - 3 - ], - type: "INT32" - } - ], - inputs: [ - "input-0" - ], - outputs: [ - "result-0" ] } ] diff --git a/operators/tensor/argmax/argmax_ERRORIF_WrongInputType_10_b_axis0/test.json b/operators/tensor/argmax/argmax_ERRORIF_WrongInputType_10_b_axis0/test.json index fe95e4715..1787f66fb 100644 --- a/operators/tensor/argmax/argmax_ERRORIF_WrongInputType_10_b_axis0/test.json +++ b/operators/tensor/argmax/argmax_ERRORIF_WrongInputType_10_b_axis0/test.json @@ -5,16 +5,43 @@ _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 + ], + type: "BOOL" + }, + { + name: "result-0", + shape: [ + + ], + type: "INT32" + } + ], inputs: [ "input-0" ], @@ -22,28 +49,6 @@ "result-0" ] } - ], - tensors: [ - { - name: "input-0", - shape: [ - 10 - ], - type: "BOOL" - }, - { - name: "result-0", - shape: [ - - ], - type: "INT32" - } - ], - inputs: [ - "input-0" - ], - outputs: [ - "result-0" ] } ] diff --git a/operators/tensor/argmax/argmax_ERRORIF_WrongInputType_10x2_i32_axis0/test.json b/operators/tensor/argmax/argmax_ERRORIF_WrongInputType_10x2_i32_axis0/test.json index 1c65abffe..0d27a5454 100644 --- a/operators/tensor/argmax/argmax_ERRORIF_WrongInputType_10x2_i32_axis0/test.json +++ b/operators/tensor/argmax/argmax_ERRORIF_WrongInputType_10x2_i32_axis0/test.json @@ -5,16 +5,44 @@ _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, + 2 + ], + type: "INT32" + }, + { + name: "result-0", + shape: [ + 2 + ], + type: "INT32" + } + ], inputs: [ "input-0" ], @@ -22,29 +50,6 @@ "result-0" ] } - ], - tensors: [ - { - name: "input-0", - shape: [ - 10, - 2 - ], - type: "INT32" - }, - { - name: "result-0", - shape: [ - 2 - ], - type: "INT32" - } - ], - inputs: [ - "input-0" - ], - outputs: [ - "result-0" ] } ] diff --git a/operators/tensor/argmax/argmax_ERRORIF_WrongOutputList_10x3_f32_axis0/test.json b/operators/tensor/argmax/argmax_ERRORIF_WrongOutputList_10x3_f32_axis0/test.json index 85269ea3d..75581f53b 100644 --- a/operators/tensor/argmax/argmax_ERRORIF_WrongOutputList_10x3_f32_axis0/test.json +++ b/operators/tensor/argmax/argmax_ERRORIF_WrongOutputList_10x3_f32_axis0/test.json @@ -5,47 +5,52 @@ _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", + "eiDummyOutput" + ] + } + ], + tensors: [ + { + name: "input-0", + shape: [ + 10, + 3 + ], + type: "FP32" + }, + { + name: "result-0", + shape: [ + 3 + ], + type: "INT32" + } + ], inputs: [ "input-0" ], outputs: [ - "result-0", - "eiDummyOutput" + "result-0" ] } - ], - tensors: [ - { - name: "input-0", - shape: [ - 10, - 3 - ], - type: "FP32" - }, - { - name: "result-0", - shape: [ - 3 - ], - type: "INT32" - } - ], - inputs: [ - "input-0" - ], - outputs: [ - "result-0" ] } ] diff --git a/operators/tensor/argmax/argmax_ERRORIF_WrongOutputList_11x10x10x6_i8_axis0/test.json b/operators/tensor/argmax/argmax_ERRORIF_WrongOutputList_11x10x10x6_i8_axis0/test.json index ca88ea416..56b1dff24 100644 --- a/operators/tensor/argmax/argmax_ERRORIF_WrongOutputList_11x10x10x6_i8_axis0/test.json +++ b/operators/tensor/argmax/argmax_ERRORIF_WrongOutputList_11x10x10x6_i8_axis0/test.json @@ -5,50 +5,55 @@ _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: [ + + ] + } + ], + tensors: [ + { + name: "input-0", + shape: [ + 11, + 10, + 10, + 6 + ], + type: "INT8" + }, + { + name: "result-0", + shape: [ + 10, + 10, + 6 + ], + type: "INT32" + } + ], inputs: [ "input-0" ], outputs: [ - + "result-0" ] } - ], - tensors: [ - { - name: "input-0", - shape: [ - 11, - 10, - 10, - 6 - ], - type: "INT8" - }, - { - name: "result-0", - shape: [ - 10, - 10, - 6 - ], - type: "INT32" - } - ], - inputs: [ - "input-0" - ], - outputs: [ - "result-0" ] } ] diff --git a/operators/tensor/argmax/argmax_ERRORIF_WrongOutputType_10x6_bf16_axis0/test.json b/operators/tensor/argmax/argmax_ERRORIF_WrongOutputType_10x6_bf16_axis0/test.json index ea58f6e46..009710485 100644 --- a/operators/tensor/argmax/argmax_ERRORIF_WrongOutputType_10x6_bf16_axis0/test.json +++ b/operators/tensor/argmax/argmax_ERRORIF_WrongOutputType_10x6_bf16_axis0/test.json @@ -5,16 +5,44 @@ _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, + 6 + ], + type: "BF16" + }, + { + name: "result-0", + shape: [ + 6 + ], + type: "INT48" + } + ], inputs: [ "input-0" ], @@ -22,29 +50,6 @@ "result-0" ] } - ], - tensors: [ - { - name: "input-0", - shape: [ - 10, - 6 - ], - type: "BF16" - }, - { - name: "result-0", - shape: [ - 6 - ], - type: "INT48" - } - ], - inputs: [ - "input-0" - ], - outputs: [ - "result-0" ] } ] diff --git a/operators/tensor/argmax/argmax_ERRORIF_WrongOutputType_5x7_i16_axis0/test.json b/operators/tensor/argmax/argmax_ERRORIF_WrongOutputType_5x7_i16_axis0/test.json index 97c4f1cb3..f12f164e8 100644 --- a/operators/tensor/argmax/argmax_ERRORIF_WrongOutputType_5x7_i16_axis0/test.json +++ b/operators/tensor/argmax/argmax_ERRORIF_WrongOutputType_5x7_i16_axis0/test.json @@ -5,16 +5,44 @@ _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: [ + 5, + 7 + ], + type: "INT16" + }, + { + name: "result-0", + shape: [ + 7 + ], + type: "FP16" + } + ], inputs: [ "input-0" ], @@ -22,29 +50,6 @@ "result-0" ] } - ], - tensors: [ - { - name: "input-0", - shape: [ - 5, - 7 - ], - type: "INT16" - }, - { - name: "result-0", - shape: [ - 7 - ], - type: "FP16" - } - ], - inputs: [ - "input-0" - ], - outputs: [ - "result-0" ] } ] diff --git a/operators/tensor/argmax/argmax_ERRORIF_WrongRank_10x11x5x11x11_i8_axis0/test.json b/operators/tensor/argmax/argmax_ERRORIF_WrongRank_10x11x5x11x11_i8_axis0/test.json index 2bc3bc214..420de1831 100644 --- a/operators/tensor/argmax/argmax_ERRORIF_WrongRank_10x11x5x11x11_i8_axis0/test.json +++ b/operators/tensor/argmax/argmax_ERRORIF_WrongRank_10x11x5x11x11_i8_axis0/test.json @@ -5,16 +5,50 @@ _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, + 5, + 11, + 11 + ], + type: "INT8" + }, + { + name: "result-0", + shape: [ + 11, + 5, + 11, + 11 + ], + type: "INT32" + } + ], inputs: [ "input-0" ], @@ -22,35 +56,6 @@ "result-0" ] } - ], - tensors: [ - { - name: "input-0", - shape: [ - 10, - 11, - 5, - 11, - 11 - ], - type: "INT8" - }, - { - name: "result-0", - shape: [ - 11, - 5, - 11, - 11 - ], - type: "INT32" - } - ], - inputs: [ - "input-0" - ], - outputs: [ - "result-0" ] } ] diff --git a/operators/tensor/argmax/argmax_ERRORIF_WrongRank_1x6x7x15x4_bf16_axis0/test.json b/operators/tensor/argmax/argmax_ERRORIF_WrongRank_1x6x7x15x4_bf16_axis0/test.json index bb19b1218..cd84f59c7 100644 --- a/operators/tensor/argmax/argmax_ERRORIF_WrongRank_1x6x7x15x4_bf16_axis0/test.json +++ b/operators/tensor/argmax/argmax_ERRORIF_WrongRank_1x6x7x15x4_bf16_axis0/test.json @@ -5,16 +5,50 @@ _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: [ + 1, + 6, + 7, + 15, + 4 + ], + type: "BF16" + }, + { + name: "result-0", + shape: [ + 6, + 7, + 15, + 4 + ], + type: "INT32" + } + ], inputs: [ "input-0" ], @@ -22,35 +56,6 @@ "result-0" ] } - ], - tensors: [ - { - name: "input-0", - shape: [ - 1, - 6, - 7, - 15, - 4 - ], - type: "BF16" - }, - { - name: "result-0", - shape: [ - 6, - 7, - 15, - 4 - ], - type: "INT32" - } - ], - inputs: [ - "input-0" - ], - outputs: [ - "result-0" ] } ] |