diff options
author | Georgios Pinitas <georgios.pinitas@arm.com> | 2019-04-03 17:01:26 +0100 |
---|---|---|
committer | Pablo Marquez <pablo.tello@arm.com> | 2019-04-04 15:51:15 +0000 |
commit | 7191aaa184992ca1fa8bdbbe5a9d8f9cd093ad5b (patch) | |
tree | bca9abe51c21e72403ae0ab99e0f0b49a8095980 /arm_compute/core/CL | |
parent | a52e4cf36ec86b63660f5a687073fa0985384dc1 (diff) | |
download | ComputeLibrary-7191aaa184992ca1fa8bdbbe5a9d8f9cd093ad5b.tar.gz |
COMPMID-2101: DenseNet issue on NHWC
Workaround data layout issue for WinogradOutputTransform.
Change-Id: If9ced178eccbb36d959862da09339ba5c52b0f6d
Signed-off-by: Georgios Pinitas <georgios.pinitas@arm.com>
Reviewed-on: https://review.mlplatform.org/c/943
Reviewed-by: Pablo Marquez <pablo.tello@arm.com>
Comments-Addressed: Arm Jenkins <bsgcomp@arm.com>
Tested-by: Arm Jenkins <bsgcomp@arm.com>
Diffstat (limited to 'arm_compute/core/CL')
-rw-r--r-- | arm_compute/core/CL/kernels/CLWinogradOutputTransformKernel.h | 3 |
1 files changed, 2 insertions, 1 deletions
diff --git a/arm_compute/core/CL/kernels/CLWinogradOutputTransformKernel.h b/arm_compute/core/CL/kernels/CLWinogradOutputTransformKernel.h index bdb230d645..2bcd292ac7 100644 --- a/arm_compute/core/CL/kernels/CLWinogradOutputTransformKernel.h +++ b/arm_compute/core/CL/kernels/CLWinogradOutputTransformKernel.h @@ -1,5 +1,5 @@ /* - * Copyright (c) 2018 ARM Limited. + * Copyright (c) 2018-2019 ARM Limited. * * SPDX-License-Identifier: MIT * @@ -99,6 +99,7 @@ private: const ICLTensor *_input; const ICLTensor *_bias; ICLTensor *_output; + bool _is_nhwc; }; } // namespace arm_compute #endif /*__ARM_COMPUTE_CLWINOGRADOUTPUTTRANSFORMKERNEL_H__ */ |