Make TopK support 1D, 2D tensor (#2323)
author윤현식/동작제어Lab(SR)/Principal Engineer/삼성전자 <hyunsik.yoon@samsung.com>
Mon, 20 Aug 2018 02:55:30 +0000 (11:55 +0900)
committer박세희/동작제어Lab(SR)/Principal Engineer/삼성전자 <saehie.park@samsung.com>
Mon, 20 Aug 2018 02:55:30 +0000 (11:55 +0900)
commitafde841a7d0c6fc0bd2caa67ccaa89501680207c
tree22ad01e8c4a94acb09d753fe8d63b24cacd098e7
parent027199610ce1e38f1a9b4b6958cd10203efc9cee
Make TopK support 1D, 2D tensor (#2323)

Previous TopK supports 1D only.
This commit makes TopK support 2D also in runtime layer.
(ACL layer PR : https://github.sec.samsung.net/RS7-RuntimeNTools/ComputeLibrary/pull/41)

Signed-off-by: Hyun Sik Yoon <hyunsik.yoon@samsung.com>
runtimes/pure_arm_compute/src/compilation.cc