diff options
Diffstat (limited to 'tests/nnapi/specs/Ex/unpack_ex_3D_int_2.mod.py')
-rw-r--r-- | tests/nnapi/specs/Ex/unpack_ex_3D_int_2.mod.py | 12 |
1 files changed, 6 insertions, 6 deletions
diff --git a/tests/nnapi/specs/Ex/unpack_ex_3D_int_2.mod.py b/tests/nnapi/specs/Ex/unpack_ex_3D_int_2.mod.py index 115954972..db51351c5 100644 --- a/tests/nnapi/specs/Ex/unpack_ex_3D_int_2.mod.py +++ b/tests/nnapi/specs/Ex/unpack_ex_3D_int_2.mod.py @@ -1,11 +1,11 @@ -# Sample UnPack model, axis = 2 +# Sample UnPack model, axis = 1 model = Model() -input = Input("input", "TENSOR_INT32", "{3, 2, 3, 4}") -axis = Int32Scalar("axis", 2) +input = Input("input", "TENSOR_INT32", "{6, 3, 4}") +axis = Int32Scalar("axis", 1) num_splits = Int32Scalar("num_splits", 3) -out1 = Output("output1", "TENSOR_INT32", "{3, 2, 4}") -out2 = Output("output2", "TENSOR_INT32", "{3, 2, 4}") -out3 = Output("output3", "TENSOR_INT32", "{3, 2, 4}") +out1 = Output("output1", "TENSOR_INT32", "{6, 4}") +out2 = Output("output2", "TENSOR_INT32", "{6, 4}") +out3 = Output("output3", "TENSOR_INT32", "{6, 4}") model = model.Operation("UNPACK_EX", input, num_splits, axis).To([out1, out2, out3]) input0 = {input: # input 0 |