diff options
| author | 오형석/On-Device Lab(SR)/Staff Engineer/삼성전자 <hseok82.oh@samsung.com> | 2019-03-08 09:50:55 +0900 |
|---|---|---|
| committer | GitHub Enterprise <noreply-CODE@samsung.com> | 2019-03-08 09:50:55 +0900 |
| commit | feb76df98badcae9caca4f61f3af2f74912334ff (patch) | |
| tree | e0ee228c4935c2e6c5888cb51fdc347c53ce3e10 /tests/nnapi/specs/Ex/split_ex_1D_float.mod.py | |
| parent | dc1ebb6b03a97090306a32181295cbe62e72a27d (diff) | |
| download | nnfw-feb76df98badcae9caca4f61f3af2f74912334ff.tar.gz nnfw-feb76df98badcae9caca4f61f3af2f74912334ff.tar.bz2 nnfw-feb76df98badcae9caca4f61f3af2f74912334ff.zip | |
Fix split implementation and test (#4622)
* Fix split implementation and test
Fix plit implementation based on NeuralNetworksEx.h spec
- Fix split in nnapi_delegate.cpp
- Fix pacl implementation
- Fix spli test
Signed-off-by: Hyeongseok Oh <hseok82.oh@samsung.com>
* Fix comment
Diffstat (limited to 'tests/nnapi/specs/Ex/split_ex_1D_float.mod.py')
| -rw-r--r-- | tests/nnapi/specs/Ex/split_ex_1D_float.mod.py | 3 |
1 files changed, 2 insertions, 1 deletions
diff --git a/tests/nnapi/specs/Ex/split_ex_1D_float.mod.py b/tests/nnapi/specs/Ex/split_ex_1D_float.mod.py index 97b61462e..137ece828 100644 --- a/tests/nnapi/specs/Ex/split_ex_1D_float.mod.py +++ b/tests/nnapi/specs/Ex/split_ex_1D_float.mod.py @@ -2,6 +2,7 @@ model = Model() i1 = Input("op1", "TENSOR_FLOAT32", "{8}") axis = Int32Scalar("axis", 0) +num_out = Int32Scalar("num_out", 8) i2 = Output("op2", "TENSOR_FLOAT32", "{1}") i3 = Output("op3", "TENSOR_FLOAT32", "{1}") i4 = Output("op4", "TENSOR_FLOAT32", "{1}") @@ -11,7 +12,7 @@ i7 = Output("op7", "TENSOR_FLOAT32", "{1}") i8 = Output("op8", "TENSOR_FLOAT32", "{1}") i9 = Output("op9", "TENSOR_FLOAT32", "{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 |