diff options
author | Michalis Spyrou <michalis.spyrou@arm.com> | 2020-03-04 17:22:55 +0000 |
---|---|---|
committer | Michele Di Giorgio <michele.digiorgio@arm.com> | 2020-03-06 15:51:43 +0000 |
commit | 3eda16adc25d44849a9eb578c0d6d068a6f0cc50 (patch) | |
tree | 3b5b9e8d84e081883bcc40e1904fe69ad1a25856 /arm_compute | |
parent | 398b8e4a0cf02b43f4469079e95b811cc1255e29 (diff) | |
download | armcl-3eda16adc25d44849a9eb578c0d6d068a6f0cc50.tar.gz armcl-3eda16adc25d44849a9eb578c0d6d068a6f0cc50.tar.bz2 armcl-3eda16adc25d44849a9eb578c0d6d068a6f0cc50.zip |
COMPMID-3179: Remove padding from NEBatchConcatenateLayerKernel
Change-Id: Ie6b03726e215a7766db45df1c65d309a26aacff2
Signed-off-by: Michalis Spyrou <michalis.spyrou@arm.com>
Reviewed-on: https://review.mlplatform.org/c/ml/ComputeLibrary/+/2833
Comments-Addressed: Arm Jenkins <bsgcomp@arm.com>
Reviewed-by: Michele Di Giorgio <michele.digiorgio@arm.com>
Tested-by: Arm Jenkins <bsgcomp@arm.com>
Diffstat (limited to 'arm_compute')
-rw-r--r-- | arm_compute/core/NEON/kernels/NEBatchConcatenateLayerKernel.h | 4 |
1 files changed, 2 insertions, 2 deletions
diff --git a/arm_compute/core/NEON/kernels/NEBatchConcatenateLayerKernel.h b/arm_compute/core/NEON/kernels/NEBatchConcatenateLayerKernel.h index e3a784702..f943744ba 100644 --- a/arm_compute/core/NEON/kernels/NEBatchConcatenateLayerKernel.h +++ b/arm_compute/core/NEON/kernels/NEBatchConcatenateLayerKernel.h @@ -1,5 +1,5 @@ /* - * Copyright (c) 2019 ARM Limited. + * Copyright (c) 2019-2020 ARM Limited. * * SPDX-License-Identifier: MIT * @@ -79,7 +79,7 @@ public: void run(const Window &window, const ThreadInfo &info) override; private: - using BatchConcatFunction = void(const ITensor *in, ITensor *out, int batch_offset, const Window &window); + using BatchConcatFunction = void(const ITensor *in, ITensor *out, unsigned int batch_offset, const Window &window); private: BatchConcatFunction *_func; |