diff options
author | Chunseok Lee <chunseok.lee@samsung.com> | 2020-03-05 15:10:09 +0900 |
---|---|---|
committer | Chunseok Lee <chunseok.lee@samsung.com> | 2020-03-05 15:22:53 +0900 |
commit | d91a039e0eda6fd70dcd22672b8ce1817c1ca50e (patch) | |
tree | 62668ec548cf31fadbbf4e99522999ad13434a25 /runtimes/pure_arm_compute/src/internal/op/Split.cc | |
parent | bd11b24234d7d43dfe05a81c520aa01ffad06e42 (diff) | |
download | nnfw-d91a039e0eda6fd70dcd22672b8ce1817c1ca50e.tar.gz nnfw-d91a039e0eda6fd70dcd22672b8ce1817c1ca50e.tar.bz2 nnfw-d91a039e0eda6fd70dcd22672b8ce1817c1ca50e.zip |
catch up to tizen_5.5 and remove unness dir
- update to tizen_5.5
- remove dirs
Diffstat (limited to 'runtimes/pure_arm_compute/src/internal/op/Split.cc')
-rw-r--r-- | runtimes/pure_arm_compute/src/internal/op/Split.cc | 13 |
1 files changed, 8 insertions, 5 deletions
diff --git a/runtimes/pure_arm_compute/src/internal/op/Split.cc b/runtimes/pure_arm_compute/src/internal/op/Split.cc index cbd863fce..6457a106a 100644 --- a/runtimes/pure_arm_compute/src/internal/op/Split.cc +++ b/runtimes/pure_arm_compute/src/internal/op/Split.cc @@ -47,13 +47,16 @@ namespace Split Param::Param(uint32_t inputCount, const uint32_t *inputs, uint32_t outputCount, const uint32_t *outputs) { - assert(inputCount == 2); + assert(inputCount == 3); // Each input should be interpreted as follows: - // 0 -> A 0-D int32 tensor, indicating the dimension along which to split. - // 1 -> An n-D tensor, specifying the tensor to be split. - axis_index = inputs[0]; - ifm_index = inputs[1]; + // 0 -> An n-D tensor, specifying the tensor to be split. + // 1 -> A 0-D int32 tensor, indicating the dimension along which to split. + // 2 -> A 0-D int32 tensor, indicating the number of outputs + // (It can be ignored on pacl becasue pacl don't support dynamic tensor shape, + // and can be used for verification only) + ifm_index = inputs[0]; + axis_index = inputs[1]; // Each output should be interpreted as follow: // [0, outputCount) -> An n-D tensor. |