diff options
Diffstat (limited to 'tests/nnapi/specs/Ex/split_ex_4D_int32_5.mod.py')
-rw-r--r-- | tests/nnapi/specs/Ex/split_ex_4D_int32_5.mod.py | 3 |
1 files changed, 2 insertions, 1 deletions
diff --git a/tests/nnapi/specs/Ex/split_ex_4D_int32_5.mod.py b/tests/nnapi/specs/Ex/split_ex_4D_int32_5.mod.py index 30e5e245d..338794855 100644 --- a/tests/nnapi/specs/Ex/split_ex_4D_int32_5.mod.py +++ b/tests/nnapi/specs/Ex/split_ex_4D_int32_5.mod.py @@ -2,9 +2,10 @@ model = Model() i1 = Input("op1", "TENSOR_INT32", "{2,2,2,2}") axis = Int32Scalar("axis", -4) # Negative axis +num_out = Int32Scalar("num_out", 2) i2 = Output("op2", "TENSOR_INT32", "{1,2,2,2}") i3 = Output("op3", "TENSOR_INT32", "{1,2,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 |