1 /*
2 * Copyright (c) 2021-2022 Arm Limited.
3 *
4 * SPDX-License-Identifier: MIT
5 *
6 * Permission is hereby granted, free of charge, to any person obtaining a copy
7 * of this software and associated documentation files (the "Software"), to
8 * deal in the Software without restriction, including without limitation the
9 * rights to use, copy, modify, merge, publish, distribute, sublicense, and/or
10 * sell copies of the Software, and to permit persons to whom the Software is
11 * furnished to do so, subject to the following conditions:
12 *
13 * The above copyright notice and this permission notice shall be included in all
14 * copies or substantial portions of the Software.
15 *
16 * THE SOFTWARE IS PROVIDED "AS IS", WITHOUT WARRANTY OF ANY KIND, EXPRESS OR
17 * IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY,
18 * FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT. IN NO EVENT SHALL THE
19 * AUTHORS OR COPYRIGHT HOLDERS BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER
20 * LIABILITY, WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING FROM,
21 * OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN THE
22 * SOFTWARE.
23 */
24 #include "src/cpu/operators/CpuActivation.h"
25
26 #include "arm_compute/runtime/NEON/NEScheduler.h"
27 #include "src/common/IOperator.h"
28 #include "src/common/utils/LegacySupport.h"
29 #include "src/common/utils/Log.h"
30 #include "src/cpu/CpuContext.h"
31 #include "src/cpu/kernels/CpuActivationKernel.h"
32
33 namespace arm_compute
34 {
35 namespace cpu
36 {
configure(const ITensorInfo * input,ITensorInfo * output,const ActivationLayerInfo & activation_info)37 void CpuActivation::configure(const ITensorInfo *input, ITensorInfo *output, const ActivationLayerInfo &activation_info)
38 {
39 ARM_COMPUTE_LOG_PARAMS(input, output, activation_info);
40 auto k = std::make_unique<kernels::CpuActivationKernel>();
41 k->configure(input, output, activation_info);
42 _kernel = std::move(k);
43 }
44
validate(const ITensorInfo * input,const ITensorInfo * output,const ActivationLayerInfo & activation_info)45 Status CpuActivation::validate(const ITensorInfo *input, const ITensorInfo *output, const ActivationLayerInfo &activation_info)
46 {
47 return kernels::CpuActivationKernel::validate(input, output, activation_info);
48 }
49
run(ITensorPack & tensors)50 void CpuActivation::run(ITensorPack &tensors)
51 {
52 ARM_COMPUTE_ERROR_ON_MSG(tensors.empty(), "No inputs provided");
53 auto split_dimension = static_cast<kernels::CpuActivationKernel *>(_kernel.get())->get_split_dimension_hint();
54 NEScheduler::get().schedule_op(_kernel.get(), split_dimension, _kernel->window(), tensors);
55 }
56
create_activation(const AclTensorDescriptor & src,const AclTensorDescriptor & dst,const AclActivationDescriptor & act,bool is_validate)57 std::tuple<IOperator *, StatusCode> CpuContext::create_activation(const AclTensorDescriptor &src, const AclTensorDescriptor &dst, const AclActivationDescriptor &act, bool is_validate)
58 {
59 TensorInfo src_info = detail::convert_to_legacy_tensor_info(src);
60 TensorInfo dst_info = detail::convert_to_legacy_tensor_info(dst);
61 auto info = detail::convert_to_activation_info(act);
62
63 if(is_validate && !bool(CpuActivation::validate(&src_info.set_is_resizable(false), &dst_info.set_is_resizable(false), info)))
64 {
65 return std::make_tuple(nullptr, StatusCode::UnsupportedConfig);
66 }
67
68 auto act_op = std::make_unique<cpu::CpuActivation>();
69 act_op->configure(&src_info, &dst_info, info);
70
71 auto op = new arm_compute::IOperator(static_cast<IContext *>(this));
72 if(op == nullptr)
73 {
74 ARM_COMPUTE_LOG_ERROR_ACL("Couldn't allocate internal resources");
75 return std::make_tuple(nullptr, StatusCode::OutOfMemory);
76 }
77 op->set_internal_operator(std::move(act_op));
78
79 return std::make_tuple(op, StatusCode::Success);
80 }
81 } // namespace cpu
82 } // namespace arm_compute
83