diff options
Diffstat (limited to 'tests')
7 files changed, 7 insertions, 7 deletions
diff --git a/tests/nnapi/specs/Ex/argmax_ex_float_1.mod.py b/tests/nnapi/specs/Ex/argmax_ex_float_1.mod.py index e2255e004..dc29fb358 100644 --- a/tests/nnapi/specs/Ex/argmax_ex_float_1.mod.py +++ b/tests/nnapi/specs/Ex/argmax_ex_float_1.mod.py @@ -1,7 +1,7 @@ model = Model() i1 = Input("input", "TENSOR_FLOAT32", "{1, 2, 2, 1}") axis = Parameter("axis", "TENSOR_INT32", "{1}", [1]) -output = Output("output", "TENSOR_INT32", "{1, 1, 2, 1}") +output = Output("output", "TENSOR_INT32", "{1, 2, 1}") model = model.Operation("ARGMAX_EX", i1, axis).To(output) diff --git a/tests/nnapi/specs/Ex/argmax_ex_float_2.mod.py b/tests/nnapi/specs/Ex/argmax_ex_float_2.mod.py index 6f06bfd0b..8b6ed565f 100644 --- a/tests/nnapi/specs/Ex/argmax_ex_float_2.mod.py +++ b/tests/nnapi/specs/Ex/argmax_ex_float_2.mod.py @@ -1,7 +1,7 @@ model = Model() i1 = Input("input", "TENSOR_FLOAT32", "{1, 2, 2, 2}") axis = Parameter("axis", "TENSOR_INT32", "{1}", [3]) -output = Output("output", "TENSOR_INT32", "{1, 2, 2, 1}") +output = Output("output", "TENSOR_INT32", "{1, 2, 2}") model = model.Operation("ARGMAX_EX", i1, axis).To(output) diff --git a/tests/nnapi/specs/Ex/argmax_ex_int32.mod.py b/tests/nnapi/specs/Ex/argmax_ex_int32.mod.py index f7f98afe2..b5d01dd85 100644 --- a/tests/nnapi/specs/Ex/argmax_ex_int32.mod.py +++ b/tests/nnapi/specs/Ex/argmax_ex_int32.mod.py @@ -1,7 +1,7 @@ model = Model() i1 = Input("input", "TENSOR_INT32", "{1, 2, 2, 1}") axis = Parameter("axis", "TENSOR_INT32", "{1}", [1]) -output = Output("output", "TENSOR_INT32", "{1, 1, 2, 1}") +output = Output("output", "TENSOR_INT32", "{1, 2, 1}") model = model.Operation("ARGMAX_EX", i1, axis).To(output) diff --git a/tests/nnapi/specs/Ex/argmax_ex_neg_axis_float.mod.py b/tests/nnapi/specs/Ex/argmax_ex_neg_axis_float.mod.py index b29cf1eb6..977cac061 100644 --- a/tests/nnapi/specs/Ex/argmax_ex_neg_axis_float.mod.py +++ b/tests/nnapi/specs/Ex/argmax_ex_neg_axis_float.mod.py @@ -1,7 +1,7 @@ model = Model() i1 = Input("input", "TENSOR_FLOAT32", "{1, 2, 4, 1}") axis = Parameter("axis", "TENSOR_INT32", "{1}", [-3]) -output = Output("output", "TENSOR_INT32", "{1, 1, 4, 1}") +output = Output("output", "TENSOR_INT32", "{1, 4, 1}") model = model.Operation("ARGMAX_EX", i1, axis).To(output) diff --git a/tests/nnapi/specs/Ex/argmax_ex_neg_axis_int32.mod.py b/tests/nnapi/specs/Ex/argmax_ex_neg_axis_int32.mod.py index ddc27b35a..9f448e08d 100644 --- a/tests/nnapi/specs/Ex/argmax_ex_neg_axis_int32.mod.py +++ b/tests/nnapi/specs/Ex/argmax_ex_neg_axis_int32.mod.py @@ -1,7 +1,7 @@ model = Model() i1 = Input("input", "TENSOR_INT32", "{1, 2, 4, 1}") axis = Parameter("axis", "TENSOR_INT32", "{1}", [-3]) -output = Output("output", "TENSOR_INT32", "{1, 1, 4, 1}") +output = Output("output", "TENSOR_INT32", "{1, 4, 1}") model = model.Operation("ARGMAX_EX", i1, axis).To(output) diff --git a/tests/nnapi/specs/Ex/argmax_ex_quant8.mod.py b/tests/nnapi/specs/Ex/argmax_ex_quant8.mod.py index fd3c5230b..c3131ef91 100644 --- a/tests/nnapi/specs/Ex/argmax_ex_quant8.mod.py +++ b/tests/nnapi/specs/Ex/argmax_ex_quant8.mod.py @@ -1,7 +1,7 @@ model = Model() i1 = Input("input", "TENSOR_QUANT8_ASYMM", "{1, 2, 2, 1}, 0.5f, 2") axis = Parameter("axis", "TENSOR_INT32", "{1}", [1]) -output = Output("output", "TENSOR_INT32", "{1, 1, 2, 1}") +output = Output("output", "TENSOR_INT32", "{1, 2, 1}") model = model.Operation("ARGMAX_EX", i1, axis).To(output) diff --git a/tests/nnapi/specs/Ex/argmax_ex_quant8_neg_axis.mod.py b/tests/nnapi/specs/Ex/argmax_ex_quant8_neg_axis.mod.py index c7ddc0e5d..9146d8f14 100644 --- a/tests/nnapi/specs/Ex/argmax_ex_quant8_neg_axis.mod.py +++ b/tests/nnapi/specs/Ex/argmax_ex_quant8_neg_axis.mod.py @@ -1,7 +1,7 @@ model = Model() i1 = Input("input", "TENSOR_QUANT8_ASYMM", "{1, 2, 4, 1}, 0.5f, 5") axis = Parameter("axis", "TENSOR_INT32", "{1}", [-3]) -output = Output("output", "TENSOR_INT32", "{1, 1, 4, 1}") +output = Output("output", "TENSOR_INT32", "{1, 4, 1}") model = model.Operation("ARGMAX_EX", i1, axis).To(output) |