diff options
Diffstat (limited to 'tests/nnapi/specs/Ex/argmax_ex_neg_axis_float.mod.py')
-rw-r--r-- | tests/nnapi/specs/Ex/argmax_ex_neg_axis_float.mod.py | 2 |
1 files changed, 1 insertions, 1 deletions
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) |