24.02.1
|
Go to the documentation of this file.
25 #ifndef SRC_CPU_KERNELS_CPUADDMULADDKERNEL
26 #define SRC_CPU_KERNELS_CPUADDMULADDKERNEL
43 using AddMulAddKernelPtr = std::add_pointer<void(
const ITensor *,
93 const char *
name()
const override;
100 AddMulAddKernelPtr _run_method{
nullptr};
void run_op(ITensorPack &tensors, const Window &window, const ThreadInfo &info) override
Execute the kernel on the passed window.
const char * name() const override
Name of the kernel.
CpuAddMulAddKernel()=default
decltype(strategy::transforms) typedef type
Interface for CPU tensor.
Activation Layer Information class.
static const std::vector< AddMulAddKernel > & get_available_kernels()
void configure(const ITensorInfo *input1, const ITensorInfo *input2, const ITensorInfo *bn_mul, const ITensorInfo *bn_add, ITensorInfo *add_output, ITensorInfo *final_output, ConvertPolicy policy, const ActivationLayerInfo &act_info)
Initialize the kernel's inputs and outputs.
std::add_pointer< bool(const DataTypeISASelectorData &data)>::type DataTypeISASelectorPtr
ARM_COMPUTE_DISALLOW_COPY_ALLOW_MOVE(CpuAddMulAddKernel)
const DataTypeISASelectorPtr is_selected
const Window & window() const
The maximum window the kernel can be executed on.
Information about executing thread and CPU.
AddMulAddKernelPtr ukernel
Describe a multidimensional execution window.
Copyright (c) 2017-2024 Arm Limited.
ConvertPolicy
Policy to handle integer overflow.
Interface for the kernel to perform addition between two tensors.
Store the tensor's metadata.
ScaleKernelInfo info(interpolation_policy, default_border_mode, PixelValue(), sampling_policy, false)
static Status validate(const ITensorInfo *input1, const ITensorInfo *input2, const ITensorInfo *bn_mul, const ITensorInfo *bn_add, const ITensorInfo *add_output, const ITensorInfo *final_output, ConvertPolicy policy, const ActivationLayerInfo &act_info)
Static function to check if given info will lead to a valid configuration.