24.02.1
|
Go to the documentation of this file.
41 auto k = std::make_unique<kernels::CpuActivationKernel>();
42 k->configure(
input, output, activation_info);
43 _kernel = std::move(k);
74 auto act_op = std::make_unique<cpu::CpuActivation>();
83 op->set_internal_operator(std::move(act_op));
SimpleTensor< float > src
void configure(const ITensorInfo *input, ITensorInfo *output, const ActivationLayerInfo &activation_info)
Configure operator for a given list of arguments.
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.
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.
#define ARM_COMPUTE_LOG_ERROR_ACL(msg)
void run(ITensorPack &tensors) override
Run the kernels contained in the function.
Activation Layer Information class.
std::tuple< IOperator *, StatusCode > create_activation(const AclTensorDescriptor &src, const AclTensorDescriptor &dst, const AclActivationDescriptor &act, bool is_validate) override
bool empty() const
Checks if pack is empty.
#define ARM_COMPUTE_ERROR_ON_MSG(cond, msg)
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.
static IScheduler & get()
Access the scheduler singleton.
ITensorInfo & set_is_resizable(bool is_resizable) override
Set the flag whether the tensor size can be changed.
Store the tensor's metadata.
TensorInfo convert_to_legacy_tensor_info(const AclTensorDescriptor &desc)
Convert a descriptor to a legacy format one.
Copyright (c) 2017-2024 Arm Limited.
Interface for the activation kernel.
TensorInfo src_info(src_shape, 1, data_type)
ActivationLayerInfo convert_to_activation_info(const AclActivationDescriptor &desc)
Convert an AclActivation descriptor to an internal one.
Base class for all async functions.
Store the tensor's metadata.
ScaleKernelInfo info(interpolation_policy, default_border_mode, PixelValue(), sampling_policy, false)
#define ARM_COMPUTE_LOG_PARAMS(...)