24.04
|
Go to the documentation of this file.
24 #ifndef SRC_CPU_OPERATORS_CPUADDMULADD
25 #define SRC_CPU_OPERATORS_CPUADDMULADD
std::vector< MemoryInfo > MemoryRequirements
void run(ITensorPack &tensors) override
Run the kernels contained in the function.
Basic function to run kernels::CpuAddMulAddKernel.
Activation Layer Information class.
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 operator's inputs and outputs.
Basic function to run kernels::CpuDequantizeKernel that dequantizes an input tensor.
Basic interface for functions which have a single async CPU kernel.
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.
Store the tensor's metadata.
Copyright (c) 2017-2024 Arm Limited.
ConvertPolicy
Policy to handle integer overflow.
Store the tensor's metadata.
experimental::MemoryRequirements workspace() const override
Return the memory requirements required by the workspace.