Compute Library
 22.11
CpuActivation.cpp
Go to the documentation of this file.
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  */
25 
27 #include "src/common/IOperator.h"
29 #include "src/common/utils/Log.h"
30 #include "src/cpu/CpuContext.h"
32 
33 namespace arm_compute
34 {
35 namespace cpu
36 {
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 
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 
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 
57 std::tuple<IOperator *, StatusCode> CpuContext::create_activation(const AclTensorDescriptor &src, const AclTensorDescriptor &dst, const AclActivationDescriptor &act, bool is_validate)
58 {
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
bool empty() const
Checks if pack is empty.
Definition: ITensorPack.cpp:80
std::tuple< IOperator *, StatusCode > create_activation(const AclTensorDescriptor &src, const AclTensorDescriptor &dst, const AclActivationDescriptor &act, bool is_validate) override
virtual void schedule_op(ICPPKernel *kernel, const Hints &hints, const Window &window, ITensorPack &tensors)=0
Runs the kernel in the same thread as the caller synchronously.
void configure(const ITensorInfo *input, ITensorInfo *output, const ActivationLayerInfo &activation_info)
Configure operator for a given list of arguments.
TensorInfo convert_to_legacy_tensor_info(const AclTensorDescriptor &desc)
Convert a descriptor to a legacy format one.
Base class for all async functions.
Definition: IOperator.h:37
Store the tensor&#39;s metadata.
Definition: ITensorInfo.h:40
#define ARM_COMPUTE_LOG_ERROR_ACL(msg)
Log an error message to the logger.
Definition: Log.h:77
Status class.
Definition: Error.h:52
Activation Layer Information class.
Definition: Types.h:1639
SimpleTensor< float > src
Definition: DFT.cpp:155
Copyright (c) 2017-2022 Arm Limited.
#define ARM_COMPUTE_ERROR_ON_MSG(cond, msg)
Definition: Error.h:456
ITensorInfo & set_is_resizable(bool is_resizable) override
Set the flag whether the tensor size can be changed.
Definition: TensorInfo.h:274
TensorInfo src_info(src_shape, 1, data_type)
void run(ITensorPack &tensors) override
Run the kernels contained in the function.
ScaleKernelInfo info(interpolation_policy, default_border_mode, PixelValue(), sampling_policy, false)
static Status validate(const ITensorInfo *input, const ITensorInfo *output, const ActivationLayerInfo &act_info)
Static function to check if given info will lead to a valid configuration.
Tensor packing service.
Definition: ITensorPack.h:39
#define ARM_COMPUTE_LOG_PARAMS(...)
Store the tensor&#39;s metadata.
Definition: TensorInfo.h:43
static Status validate(const ITensorInfo *src, const ITensorInfo *dst, const ActivationLayerInfo &act_info)
Static function to check if given info will lead to a valid configuration.
Interface for the activation kernel.
ActivationLayerInfo convert_to_activation_info(const AclActivationDescriptor &desc)
Convert an AclActivation descriptor to an internal one.
static IScheduler & get()
Access the scheduler singleton.
Definition: Scheduler.cpp:94