diff options
Diffstat (limited to 'src/cpu/operators')
-rw-r--r-- | src/cpu/operators/CpuConcatenate.h | 4 | ||||
-rw-r--r-- | src/cpu/operators/CpuSoftmax.h | 6 |
2 files changed, 5 insertions, 5 deletions
diff --git a/src/cpu/operators/CpuConcatenate.h b/src/cpu/operators/CpuConcatenate.h index 001ac68162..eb11926b48 100644 --- a/src/cpu/operators/CpuConcatenate.h +++ b/src/cpu/operators/CpuConcatenate.h @@ -1,5 +1,5 @@ /* - * Copyright (c) 2021 Arm Limited. + * Copyright (c) 2021-2022 Arm Limited. * * SPDX-License-Identifier: MIT * @@ -67,7 +67,7 @@ public: void run(ITensorPack &tensors) override; private: - std::vector<std::unique_ptr<ICpuKernel>> _concat_kernels{}; + std::vector<std::unique_ptr<ICPPKernel>> _concat_kernels{}; unsigned int _num_srcs{ 0 }; unsigned int _axis{ 0 }; }; 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; |