diff options
Diffstat (limited to 'frameworks/tflite/operators/split_1x4x8x19_i32_splitv_axis2')
3 files changed, 203 insertions, 59 deletions
diff --git a/frameworks/tflite/operators/split_1x4x8x19_i32_splitv_axis2/Conformance-TosaOutput_0.json b/frameworks/tflite/operators/split_1x4x8x19_i32_splitv_axis2/Conformance-TosaOutput_0.json index bae8ff4d5..075a5d1e3 100644 --- a/frameworks/tflite/operators/split_1x4x8x19_i32_splitv_axis2/Conformance-TosaOutput_0.json +++ b/frameworks/tflite/operators/split_1x4x8x19_i32_splitv_axis2/Conformance-TosaOutput_0.json @@ -1,8 +1,9 @@ { "type": "int32", "data": [ - 108, - 1195, + 648, + 508, + 147, -1485 ] }
\ No newline at end of file diff --git a/frameworks/tflite/operators/split_1x4x8x19_i32_splitv_axis2/model-tflite.json b/frameworks/tflite/operators/split_1x4x8x19_i32_splitv_axis2/model-tflite.json index 930697db4..488718080 100644 --- a/frameworks/tflite/operators/split_1x4x8x19_i32_splitv_axis2/model-tflite.json +++ b/frameworks/tflite/operators/split_1x4x8x19_i32_splitv_axis2/model-tflite.json @@ -51,7 +51,7 @@ }, { shape: [ - 3 + 4 ], type: "INT32", buffer: 3, @@ -79,7 +79,7 @@ shape: [ 1, 4, - 3, + 5, 19 ], type: "INT32", @@ -95,7 +95,7 @@ shape: [ 1, 4, - 4, + 1, 19 ], type: "INT32", @@ -125,11 +125,14 @@ }, { shape: [ - + 1, + 4, + 1, + 19 ], type: "INT32", buffer: 8, - name: "reduce_0", + name: "split3", quantization: { details_type: "NONE", quantized_dimension: 0 @@ -142,7 +145,7 @@ ], type: "INT32", buffer: 9, - name: "reduce_1", + name: "reduce_0", quantization: { details_type: "NONE", quantized_dimension: 0 @@ -155,7 +158,7 @@ ], type: "INT32", buffer: 10, - name: "reduce_2", + name: "reduce_1", quantization: { details_type: "NONE", quantized_dimension: 0 @@ -164,10 +167,36 @@ }, { shape: [ - 3 + ], type: "INT32", buffer: 11, + name: "reduce_2", + quantization: { + details_type: "NONE", + quantized_dimension: 0 + }, + is_variable: false + }, + { + shape: [ + + ], + type: "INT32", + buffer: 12, + name: "reduce_3", + quantization: { + details_type: "NONE", + quantized_dimension: 0 + }, + is_variable: false + }, + { + shape: [ + 4 + ], + type: "INT32", + buffer: 13, name: "PartitionedCall:0", quantization: { details_type: "NONE", @@ -180,7 +209,7 @@ 0 ], outputs: [ - 10 + 12 ], operators: [ { @@ -193,11 +222,12 @@ outputs: [ 4, 5, - 6 + 6, + 7 ], builtin_options_type: "SplitVOptions", builtin_options: { - num_splits: 3 + num_splits: 4 }, custom_options_format: "FLEXBUFFERS" }, @@ -208,7 +238,7 @@ 1 ], outputs: [ - 7 + 8 ], builtin_options_type: "ReducerOptions", builtin_options: { @@ -223,7 +253,7 @@ 1 ], outputs: [ - 8 + 9 ], builtin_options_type: "ReducerOptions", builtin_options: { @@ -238,7 +268,7 @@ 1 ], outputs: [ - 9 + 10 ], builtin_options_type: "ReducerOptions", builtin_options: { @@ -247,18 +277,34 @@ custom_options_format: "FLEXBUFFERS" }, { - opcode_index: 2, + opcode_index: 1, inputs: [ 7, + 1 + ], + outputs: [ + 11 + ], + builtin_options_type: "ReducerOptions", + builtin_options: { + keep_dims: false + }, + custom_options_format: "FLEXBUFFERS" + }, + { + opcode_index: 2, + inputs: [ 8, - 9 + 9, + 10, + 11 ], outputs: [ - 10 + 12 ], builtin_options_type: "PackOptions", builtin_options: { - values_count: 3, + values_count: 4, axis: 0 }, custom_options_format: "FLEXBUFFERS" @@ -295,11 +341,15 @@ }, { data: [ - 3, + 5, 0, 0, 0, - 4, + 1, + 0, + 0, + 0, + 1, 0, 0, 0, @@ -332,6 +382,10 @@ { }, { + }, + { + }, + { data: [ 49, 46, @@ -432,7 +486,7 @@ 50, 46, 49, - 49, + 50, 46, 48, 0, @@ -443,11 +497,11 @@ metadata: [ { name: "min_runtime_version", - buffer: 12 + buffer: 14 }, { name: "CONVERSION_METADATA", - buffer: 13 + buffer: 15 } ], signature_defs: [ @@ -461,7 +515,7 @@ outputs: [ { name: "output_0", - tensor_index: 10 + tensor_index: 12 } ], method_name: "serving_default", diff --git a/frameworks/tflite/operators/split_1x4x8x19_i32_splitv_axis2/test_split_1x4x8x19_i32_splitv_axis2.json b/frameworks/tflite/operators/split_1x4x8x19_i32_splitv_axis2/test_split_1x4x8x19_i32_splitv_axis2.json index 4dc9df4a6..9485a0bef 100644 --- a/frameworks/tflite/operators/split_1x4x8x19_i32_splitv_axis2/test_split_1x4x8x19_i32_splitv_axis2.json +++ b/frameworks/tflite/operators/split_1x4x8x19_i32_splitv_axis2/test_split_1x4x8x19_i32_splitv_axis2.json @@ -1,9 +1,9 @@ { version: { _major: 0, - _minor: 51, + _minor: 60, _patch: 0, - _draft: true + _draft: false }, regions: [ { @@ -25,7 +25,7 @@ size: [ 1, 4, - 3, + 5, 19 ] }, @@ -43,13 +43,13 @@ start: [ 0, 0, - 3, + 5, 0 ], size: [ 1, 4, - 4, + 1, 19 ] }, @@ -67,7 +67,7 @@ start: [ 0, 0, - 7, + 6, 0 ], size: [ @@ -85,13 +85,24 @@ ] }, { - op: "REDUCE_SUM", - attribute_type: "AxisAttribute", + op: "SLICE", + attribute_type: "SliceAttribute", attribute: { - axis: 1 + start: [ + 0, + 0, + 7, + 0 + ], + size: [ + 1, + 4, + 1, + 19 + ] }, inputs: [ - "layer_0" + "TosaInput_0" ], outputs: [ "layer_3" @@ -101,10 +112,10 @@ op: "REDUCE_SUM", attribute_type: "AxisAttribute", attribute: { - axis: 2 + axis: 1 }, inputs: [ - "layer_3" + "layer_0" ], outputs: [ "layer_4" @@ -114,7 +125,7 @@ op: "REDUCE_SUM", attribute_type: "AxisAttribute", attribute: { - axis: 3 + axis: 2 }, inputs: [ "layer_4" @@ -127,10 +138,10 @@ op: "REDUCE_SUM", attribute_type: "AxisAttribute", attribute: { - axis: 1 + axis: 3 }, inputs: [ - "layer_1" + "layer_5" ], outputs: [ "layer_6" @@ -140,10 +151,10 @@ op: "REDUCE_SUM", attribute_type: "AxisAttribute", attribute: { - axis: 2 + axis: 1 }, inputs: [ - "layer_6" + "layer_1" ], outputs: [ "layer_7" @@ -189,6 +200,32 @@ ] }, { + op: "REDUCE_SUM", + attribute_type: "AxisAttribute", + attribute: { + axis: 1 + }, + inputs: [ + "layer_3" + ], + outputs: [ + "layer_11" + ] + }, + { + op: "REDUCE_SUM", + attribute_type: "AxisAttribute", + attribute: { + axis: 3 + }, + inputs: [ + "layer_11" + ], + outputs: [ + "layer_12" + ] + }, + { op: "RESHAPE", attribute_type: "ReshapeAttribute", attribute: { @@ -197,10 +234,10 @@ ] }, inputs: [ - "layer_5" + "layer_6" ], outputs: [ - "layer_11" + "layer_13" ] }, { @@ -215,7 +252,7 @@ "layer_8" ], outputs: [ - "layer_12" + "layer_14" ] }, { @@ -230,7 +267,22 @@ "layer_10" ], outputs: [ - "layer_13" + "layer_15" + ] + }, + { + op: "RESHAPE", + attribute_type: "ReshapeAttribute", + attribute: { + new_shape: [ + 1 + ] + }, + inputs: [ + "layer_12" + ], + outputs: [ + "layer_16" ] }, { @@ -240,9 +292,10 @@ axis: 0 }, inputs: [ - "layer_11", - "layer_12", - "layer_13" + "layer_13", + "layer_14", + "layer_15", + "layer_16" ], outputs: [ "TosaOutput_0" @@ -266,7 +319,7 @@ { name: "TosaOutput_0", shape: [ - 3 + 4 ], type: "INT32", data: [ @@ -278,7 +331,7 @@ shape: [ 1, 4, - 3, + 5, 19 ], type: "INT32", @@ -291,7 +344,7 @@ shape: [ 1, 4, - 4, + 1, 19 ], type: "INT32", @@ -315,7 +368,10 @@ { name: "layer_11", shape: [ - 1 + 1, + 1, + 1, + 19 ], type: "INT32", data: [ @@ -325,6 +381,9 @@ { name: "layer_12", shape: [ + 1, + 1, + 1, 1 ], type: "INT32", @@ -343,6 +402,36 @@ ] }, { + name: "layer_14", + shape: [ + 1 + ], + type: "INT32", + data: [ + + ] + }, + { + name: "layer_15", + shape: [ + 1 + ], + type: "INT32", + data: [ + + ] + }, + { + name: "layer_16", + shape: [ + 1 + ], + type: "INT32", + data: [ + + ] + }, + { name: "layer_2", shape: [ 1, @@ -359,8 +448,8 @@ name: "layer_3", shape: [ 1, + 4, 1, - 3, 19 ], type: "INT32", @@ -373,7 +462,7 @@ shape: [ 1, 1, - 1, + 5, 19 ], type: "INT32", @@ -387,7 +476,7 @@ 1, 1, 1, - 1 + 19 ], type: "INT32", data: [ @@ -399,8 +488,8 @@ shape: [ 1, 1, - 4, - 19 + 1, + 1 ], type: "INT32", data: [ |