diff options
author | George Wort <george.wort@arm.com> | 2019-01-16 11:39:57 +0000 |
---|---|---|
committer | George Wort <george.wort@arm.com> | 2019-01-23 15:15:20 +0000 |
commit | dc9b52867b26e59ba34365f53820ddd7edf01ded (patch) | |
tree | f5fbd33c9bc04c1c46a37b975e5d07e3045e1b34 /arm_compute | |
parent | 1d4f3853dfd16f55338d772ad757db0ee8710d78 (diff) | |
download | ComputeLibrary-dc9b52867b26e59ba34365f53820ddd7edf01ded.tar.gz |
COMPMID-1795: Add support for NHWC in CLGenerateProposalsLayer
Change-Id: Idd805513783fa1323e239eac8a899d8ab04fb14f
Reviewed-on: https://review.mlplatform.org/536
Tested-by: Arm Jenkins <bsgcomp@arm.com>
Reviewed-by: Michele Di Giorgio <michele.digiorgio@arm.com>
Diffstat (limited to 'arm_compute')
-rw-r--r-- | arm_compute/runtime/CL/functions/CLGenerateProposalsLayer.h | 4 |
1 files changed, 3 insertions, 1 deletions
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; |