diff options
Diffstat (limited to 'tests/nnapi/specs/Ex/unpack_ex_3D_int_1.mod.py')
| -rw-r--r-- | tests/nnapi/specs/Ex/unpack_ex_3D_int_1.mod.py | 8 |
1 files changed, 4 insertions, 4 deletions
diff --git a/tests/nnapi/specs/Ex/unpack_ex_3D_int_1.mod.py b/tests/nnapi/specs/Ex/unpack_ex_3D_int_1.mod.py index 2a668cf7e..34e153bfc 100644 --- a/tests/nnapi/specs/Ex/unpack_ex_3D_int_1.mod.py +++ b/tests/nnapi/specs/Ex/unpack_ex_3D_int_1.mod.py @@ -1,11 +1,11 @@ # Sample UnPack model, axis = 0 model = Model() -input = Input("input", "TENSOR_INT32", "{3, 2, 3, 4}") +input = Input("input", "TENSOR_INT32", "{3, 6, 4}") axis = Int32Scalar("axis", 0) num_splits = Int32Scalar("num_splits", 3) -out1 = Output("output1", "TENSOR_INT32", "{2, 3, 4}") -out2 = Output("output2", "TENSOR_INT32", "{2, 3, 4}") -out3 = Output("output3", "TENSOR_INT32", "{2, 3, 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 |