diff options
| author | Yair Schwarzbaum <yair.schwarzbaum@arm.com> | 2022-01-12 16:38:58 +0200 |
|---|---|---|
| committer | Yair Schwarzbaum <yair.schwarzbaum@arm.com> | 2022-02-01 09:58:37 +0000 |
| commit | 46d44d26183d835d209d7ef1b9023e217dd4019d (patch) | |
| tree | 0bd403aa952b52f715a9c07bfd8a2a5906712862 /src/cpu/operators/CpuSoftmax.h | |
| parent | 0ef2c2176fd99319342a4174e15c0263ede236cd (diff) | |
| download | ComputeLibrary-46d44d26183d835d209d7ef1b9023e217dd4019d.tar.gz | |
Enable kernel selection testing (Phase #2)
Resolves COMPMID-4987
Change-Id: I1201ca3eae107989d13b6a2c6d9560de24fe112d
Signed-off-by: Yair Schwarzbaum <yair.schwarzbaum@arm.com>
Reviewed-on: https://review.mlplatform.org/c/ml/ComputeLibrary/+/7015
Tested-by: Arm Jenkins <bsgcomp@arm.com>
Reviewed-by: Giorgio Arena <giorgio.arena@arm.com>
Comments-Addressed: Arm Jenkins <bsgcomp@arm.com>
Diffstat (limited to 'src/cpu/operators/CpuSoftmax.h')
| -rw-r--r-- | src/cpu/operators/CpuSoftmax.h | 6 |
1 files changed, 3 insertions, 3 deletions
diff --git a/src/cpu/operators/CpuSoftmax.h b/src/cpu/operators/CpuSoftmax.h index 20f3f006d3..64df8704f9 100644 --- a/src/cpu/operators/CpuSoftmax.h +++ b/src/cpu/operators/CpuSoftmax.h @@ -1,5 +1,5 @@ /* - * Copyright (c) 2021 Arm Limited. + * Copyright (c) 2021-2022 Arm Limited. * * SPDX-License-Identifier: MIT * @@ -92,8 +92,8 @@ private: CpuPermute _permute_input; CpuPermute _permute_output; - std::unique_ptr<ICpuKernel> _max_kernel; - std::unique_ptr<ICpuKernel> _softmax_kernel; + std::unique_ptr<ICPPKernel> _max_kernel; + std::unique_ptr<ICPPKernel> _softmax_kernel; TensorInfo _max; TensorInfo _tmp; |