From dc9b52867b26e59ba34365f53820ddd7edf01ded Mon Sep 17 00:00:00 2001 From: George Wort Date: Wed, 16 Jan 2019 11:39:57 +0000 Subject: COMPMID-1795: Add support for NHWC in CLGenerateProposalsLayer Change-Id: Idd805513783fa1323e239eac8a899d8ab04fb14f Reviewed-on: https://review.mlplatform.org/536 Tested-by: Arm Jenkins Reviewed-by: Michele Di Giorgio --- arm_compute/runtime/CL/functions/CLGenerateProposalsLayer.h | 4 +++- 1 file changed, 3 insertions(+), 1 deletion(-) (limited to 'arm_compute') diff --git a/arm_compute/runtime/CL/functions/CLGenerateProposalsLayer.h b/arm_compute/runtime/CL/functions/CLGenerateProposalsLayer.h index 343229fe10..02f5bd1ea8 100644 --- a/arm_compute/runtime/CL/functions/CLGenerateProposalsLayer.h +++ b/arm_compute/runtime/CL/functions/CLGenerateProposalsLayer.h @@ -1,5 +1,5 @@ /* - * Copyright (c) 2018 ARM Limited. + * Copyright (c) 2018-2019 ARM Limited. * * SPDX-License-Identifier: MIT * @@ -124,6 +124,8 @@ private: // CPP kernels CPPBoxWithNonMaximaSuppressionLimitKernel _cpp_nms_kernel; + bool _is_nhwc; + // Temporary tensors CLTensor _deltas_permuted; CLTensor _deltas_flattened; -- cgit v1.2.1