diff options
Diffstat (limited to 'tests/nnapi/specs/Ex/split_ex_4D_int32_2.mod.py')
| -rw-r--r-- | tests/nnapi/specs/Ex/split_ex_4D_int32_2.mod.py | 3 |
1 files changed, 2 insertions, 1 deletions
diff --git a/tests/nnapi/specs/Ex/split_ex_4D_int32_2.mod.py b/tests/nnapi/specs/Ex/split_ex_4D_int32_2.mod.py index 83253cb14..88a7cca48 100644 --- a/tests/nnapi/specs/Ex/split_ex_4D_int32_2.mod.py +++ b/tests/nnapi/specs/Ex/split_ex_4D_int32_2.mod.py @@ -2,9 +2,10 @@ model = Model() i1 = Input("op1", "TENSOR_INT32", "{2,2,2,2}") axis = Int32Scalar("axis", 1) +num_out = Int32Scalar("num_out", 2) i2 = Output("op2", "TENSOR_INT32", "{2,1,2,2}") i3 = Output("op3", "TENSOR_INT32", "{2,1,2,2}") -model = model.Operation("SPLIT_EX", axis, i1).To([i2, i3]) +model = model.Operation("SPLIT_EX", i1, axis, num_out).To([i2, i3]) # Example 1. Input in operand 0, input0 = {i1: # input 0 |