diff options
author | Yair Schwarzbaum <yair.schwarzbaum@arm.com> | 2021-11-08 10:58:06 +0200 |
---|---|---|
committer | Yair Schwarzbaum <yair.schwarzbaum@arm.com> | 2021-11-23 19:14:24 +0000 |
commit | 9e8a7701532377073220613b3642468f96cba62f (patch) | |
tree | dbbebcb6cd5e1f720eb40a462813147706fbe4dc /src/core/NEON/kernels/NERangeKernel.h | |
parent | 9cca592c13f1e688a35698641069bcd37a525f0c (diff) | |
download | ComputeLibrary-9e8a7701532377073220613b3642468f96cba62f.tar.gz |
Decouple data type for NERangeKernel
- Decouple data type for CPU implementation,
supported data types are: fp32, fp16, u8, u16, u32, s8, s16, s32
Resolves COMPMID-4612
Signed-off-by: Yair Schwarzbaum <yair.schwarzbaum@arm.com>
Change-Id: Iec9aab9f59c5a344950c788281fc500290a19bbc
Reviewed-on: https://review.mlplatform.org/c/ml/ComputeLibrary/+/6686
Tested-by: Arm Jenkins <bsgcomp@arm.com>
Comments-Addressed: Arm Jenkins <bsgcomp@arm.com>
Reviewed-by: Giorgio Arena <giorgio.arena@arm.com>
Diffstat (limited to 'src/core/NEON/kernels/NERangeKernel.h')
-rw-r--r-- | src/core/NEON/kernels/NERangeKernel.h | 11 |
1 files changed, 5 insertions, 6 deletions
diff --git a/src/core/NEON/kernels/NERangeKernel.h b/src/core/NEON/kernels/NERangeKernel.h index 7c42ef11dc..90560995e6 100644 --- a/src/core/NEON/kernels/NERangeKernel.h +++ b/src/core/NEON/kernels/NERangeKernel.h @@ -1,5 +1,5 @@ /* - * Copyright (c) 2018-2020 Arm Limited. + * Copyright (c) 2018-2021 Arm Limited. * * SPDX-License-Identifier: MIT * @@ -80,11 +80,10 @@ public: private: using RangeFunction = void(ITensor *output, float start, float step, const Window &window); - RangeFunction *_func; /**< Range function to be called */ - float _start; /**< Start of sequence */ - float _end; /**< End of sequence */ - float _step; /**< Increment/step value */ - ITensor *_output; /**< Destination tensor */ + float _start; /**< Start of sequence */ + float _end; /**< End of sequence */ + float _step; /**< Increment/step value */ + ITensor *_output; /**< Destination tensor */ }; } // namespace arm_compute #endif /* ARM_COMPUTE_NERANGEKERNEL_H */ |