diff options
Diffstat (limited to 'frameworks/tflite/operators/split_13x21x3_i32_splitv_axis0/test_split_13x21x3_i32_splitv_axis0.json')
| -rw-r--r-- | frameworks/tflite/operators/split_13x21x3_i32_splitv_axis0/test_split_13x21x3_i32_splitv_axis0.json | 204 |
1 files changed, 157 insertions, 47 deletions
diff --git a/frameworks/tflite/operators/split_13x21x3_i32_splitv_axis0/test_split_13x21x3_i32_splitv_axis0.json b/frameworks/tflite/operators/split_13x21x3_i32_splitv_axis0/test_split_13x21x3_i32_splitv_axis0.json index 8ae4ee1bd..92e8394d1 100644 --- a/frameworks/tflite/operators/split_13x21x3_i32_splitv_axis0/test_split_13x21x3_i32_splitv_axis0.json +++ b/frameworks/tflite/operators/split_13x21x3_i32_splitv_axis0/test_split_13x21x3_i32_splitv_axis0.json @@ -22,7 +22,7 @@ 0 ], size: [ - 3, + 6, 21, 3 ] @@ -39,12 +39,12 @@ attribute_type: "SliceAttribute", attribute: { start: [ - 3, + 6, 0, 0 ], size: [ - 7, + 1, 21, 3 ] @@ -61,7 +61,7 @@ attribute_type: "SliceAttribute", attribute: { start: [ - 10, + 7, 0, 0 ], @@ -83,12 +83,12 @@ attribute_type: "SliceAttribute", attribute: { start: [ - 11, + 8, 0, 0 ], size: [ - 1, + 3, 21, 3 ] @@ -105,7 +105,7 @@ attribute_type: "SliceAttribute", attribute: { start: [ - 12, + 11, 0, 0 ], @@ -123,13 +123,22 @@ ] }, { - op: "REDUCE_SUM", - attribute_type: "AxisAttribute", + op: "SLICE", + attribute_type: "SliceAttribute", attribute: { - axis: 0 + start: [ + 12, + 0, + 0 + ], + size: [ + 1, + 21, + 3 + ] }, inputs: [ - "layer_0" + "TosaInput_0" ], outputs: [ "layer_5" @@ -139,10 +148,10 @@ op: "REDUCE_SUM", attribute_type: "AxisAttribute", attribute: { - axis: 1 + axis: 0 }, inputs: [ - "layer_5" + "layer_0" ], outputs: [ "layer_6" @@ -152,7 +161,7 @@ op: "REDUCE_SUM", attribute_type: "AxisAttribute", attribute: { - axis: 2 + axis: 1 }, inputs: [ "layer_6" @@ -165,10 +174,10 @@ op: "REDUCE_SUM", attribute_type: "AxisAttribute", attribute: { - axis: 0 + axis: 2 }, inputs: [ - "layer_1" + "layer_7" ], outputs: [ "layer_8" @@ -181,7 +190,7 @@ axis: 1 }, inputs: [ - "layer_8" + "layer_1" ], outputs: [ "layer_9" @@ -230,7 +239,7 @@ op: "REDUCE_SUM", attribute_type: "AxisAttribute", attribute: { - axis: 1 + axis: 0 }, inputs: [ "layer_3" @@ -243,7 +252,7 @@ op: "REDUCE_SUM", attribute_type: "AxisAttribute", attribute: { - axis: 2 + axis: 1 }, inputs: [ "layer_13" @@ -256,13 +265,26 @@ op: "REDUCE_SUM", attribute_type: "AxisAttribute", attribute: { + axis: 2 + }, + inputs: [ + "layer_14" + ], + outputs: [ + "layer_15" + ] + }, + { + op: "REDUCE_SUM", + attribute_type: "AxisAttribute", + attribute: { axis: 1 }, inputs: [ "layer_4" ], outputs: [ - "layer_15" + "layer_16" ] }, { @@ -272,10 +294,36 @@ axis: 2 }, inputs: [ - "layer_15" + "layer_16" ], outputs: [ - "layer_16" + "layer_17" + ] + }, + { + op: "REDUCE_SUM", + attribute_type: "AxisAttribute", + attribute: { + axis: 1 + }, + inputs: [ + "layer_5" + ], + outputs: [ + "layer_18" + ] + }, + { + op: "REDUCE_SUM", + attribute_type: "AxisAttribute", + attribute: { + axis: 2 + }, + inputs: [ + "layer_18" + ], + outputs: [ + "layer_19" ] }, { @@ -287,10 +335,10 @@ ] }, inputs: [ - "layer_7" + "layer_8" ], outputs: [ - "layer_17" + "layer_20" ] }, { @@ -305,7 +353,7 @@ "layer_10" ], outputs: [ - "layer_18" + "layer_21" ] }, { @@ -320,7 +368,7 @@ "layer_12" ], outputs: [ - "layer_19" + "layer_22" ] }, { @@ -332,10 +380,10 @@ ] }, inputs: [ - "layer_14" + "layer_15" ], outputs: [ - "layer_20" + "layer_23" ] }, { @@ -347,10 +395,25 @@ ] }, inputs: [ - "layer_16" + "layer_17" ], outputs: [ - "layer_21" + "layer_24" + ] + }, + { + op: "RESHAPE", + attribute_type: "ReshapeAttribute", + attribute: { + new_shape: [ + 1 + ] + }, + inputs: [ + "layer_19" + ], + outputs: [ + "layer_25" ] }, { @@ -360,11 +423,12 @@ axis: 0 }, inputs: [ - "layer_17", - "layer_18", - "layer_19", "layer_20", - "layer_21" + "layer_21", + "layer_22", + "layer_23", + "layer_24", + "layer_25" ], outputs: [ "TosaOutput_0" @@ -387,7 +451,7 @@ { name: "TosaOutput_0", shape: [ - 5 + 6 ], type: "INT32", data: [ @@ -397,7 +461,7 @@ { name: "layer_0", shape: [ - 3, + 6, 21, 3 ], @@ -409,7 +473,7 @@ { name: "layer_1", shape: [ - 7, + 1, 21, 3 ], @@ -458,7 +522,7 @@ name: "layer_13", shape: [ 1, - 1, + 21, 3 ], type: "INT32", @@ -471,7 +535,7 @@ shape: [ 1, 1, - 1 + 3 ], type: "INT32", data: [ @@ -483,7 +547,7 @@ shape: [ 1, 1, - 3 + 1 ], type: "INT32", data: [ @@ -495,7 +559,7 @@ shape: [ 1, 1, - 1 + 3 ], type: "INT32", data: [ @@ -505,6 +569,8 @@ { name: "layer_17", shape: [ + 1, + 1, 1 ], type: "INT32", @@ -515,7 +581,9 @@ { name: "layer_18", shape: [ - 1 + 1, + 1, + 3 ], type: "INT32", data: [ @@ -525,6 +593,8 @@ { name: "layer_19", shape: [ + 1, + 1, 1 ], type: "INT32", @@ -565,9 +635,49 @@ ] }, { + name: "layer_22", + shape: [ + 1 + ], + type: "INT32", + data: [ + + ] + }, + { + name: "layer_23", + shape: [ + 1 + ], + type: "INT32", + data: [ + + ] + }, + { + name: "layer_24", + shape: [ + 1 + ], + type: "INT32", + data: [ + + ] + }, + { + name: "layer_25", + shape: [ + 1 + ], + type: "INT32", + data: [ + + ] + }, + { name: "layer_3", shape: [ - 1, + 3, 21, 3 ], @@ -604,7 +714,7 @@ name: "layer_6", shape: [ 1, - 1, + 21, 3 ], type: "INT32", @@ -617,7 +727,7 @@ shape: [ 1, 1, - 1 + 3 ], type: "INT32", data: [ @@ -628,8 +738,8 @@ name: "layer_8", shape: [ 1, - 21, - 3 + 1, + 1 ], type: "INT32", data: [ |