diff options
Diffstat (limited to 'tests/nnapi/specs/Ex/split_ex_1D_int32.mod.py')
-rw-r--r-- | tests/nnapi/specs/Ex/split_ex_1D_int32.mod.py | 3 |
1 files changed, 2 insertions, 1 deletions
diff --git a/tests/nnapi/specs/Ex/split_ex_1D_int32.mod.py b/tests/nnapi/specs/Ex/split_ex_1D_int32.mod.py index 378b37707..9bfc215f6 100644 --- a/tests/nnapi/specs/Ex/split_ex_1D_int32.mod.py +++ b/tests/nnapi/specs/Ex/split_ex_1D_int32.mod.py @@ -2,6 +2,7 @@ model = Model() i1 = Input("op1", "TENSOR_INT32", "{8}") axis = Int32Scalar("axis", 0) +num_out = Int32Scalar("num_out", 8) i2 = Output("op2", "TENSOR_INT32", "{1}") i3 = Output("op3", "TENSOR_INT32", "{1}") i4 = Output("op4", "TENSOR_INT32", "{1}") @@ -11,7 +12,7 @@ i7 = Output("op7", "TENSOR_INT32", "{1}") i8 = Output("op8", "TENSOR_INT32", "{1}") i9 = Output("op9", "TENSOR_INT32", "{1}") -model = model.Operation("SPLIT_EX", axis, i1).To([i2, i3, i4, i5, i6, i7, i8, i9]) +model = model.Operation("SPLIT_EX", i1, axis, num_out).To([i2, i3, i4, i5, i6, i7, i8, i9]) # Example 1. Input in operand 0, input0 = {i1: # input 0 |