From 7fea1ad478f187f6d9cff5ff7da32aa412532883 Mon Sep 17 00:00:00 2001 From: =?utf8?q?=EC=98=A4=ED=98=95=EC=84=9D/=EB=8F=99=EC=9E=91=EC=A0=9C?= =?utf8?q?=EC=96=B4Lab=28SR=29/Staff=20Engineer/=EC=82=BC=EC=84=B1?= =?utf8?q?=EC=A0=84=EC=9E=90?= Date: Tue, 14 Aug 2018 13:39:28 +0900 Subject: [PATCH] [neurun] Fix feature to ACL TensorShape conversion (#2290) Fix feature to ACL TensorShape conversion to set batch value Signed-off-by: Hyeongseok Oh --- runtimes/neurun/src/internal/Convert.cc | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/runtimes/neurun/src/internal/Convert.cc b/runtimes/neurun/src/internal/Convert.cc index fa577f9..3c241fc 100644 --- a/runtimes/neurun/src/internal/Convert.cc +++ b/runtimes/neurun/src/internal/Convert.cc @@ -10,7 +10,7 @@ namespace internal ::arm_compute::TensorShape asTensorShape(const nnfw::util::feature::Shape &shape) { - return ::arm_compute::TensorShape(shape.W, shape.H, shape.C, 1); + return ::arm_compute::TensorShape(shape.W, shape.H, shape.C, shape.N); } ::arm_compute::TensorShape asTensorShape(const nnfw::util::kernel::Shape &shape) -- 2.7.4