24.04
|
Go to the documentation of this file.
35 struct NEAddMulAdd::Impl
37 std::unique_ptr<cpu::CpuAddMulAdd> op{
nullptr};
38 WorkspaceData<Tensor> workspace_tensors{};
39 ITensorPack run_pack{};
40 MemoryGroup memory_group{};
45 _impl->memory_group =
MemoryGroup(std::move(memory_manager));
61 _impl->op = std::make_unique<cpu::CpuAddMulAdd>();
62 _impl->op->configure(input1->
info(), input2->
info(), bn_mul->
info(), bn_add->
info(),
63 add_output !=
nullptr ? add_output->
info() :
nullptr, final_output->
info(), policy,
act_info);
70 _impl->workspace_tensors = manage_workspace<Tensor>(_impl->op->workspace(), _impl->memory_group, _impl->run_pack);
87 _impl->op->run(_impl->run_pack);
Interface for CPU tensor.
~NEAddMulAdd()
Destructor.
Activation Layer Information class.
virtual ITensorInfo * info() const =0
Interface to be implemented by the child class to return the tensor's metadata.
void run() override
Run the kernels contained in the function.
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.
Copyright (c) 2017-2024 Arm Limited.
ConvertPolicy
Policy to handle integer overflow.
NEAddMulAdd(std::shared_ptr< IMemoryManager > memory_manager=nullptr)
Constructor.
void configure(ITensor *input1, ITensor *input2, ITensor *bn_mul, ITensor *bn_add, ITensor *add_output, ITensor *final_output, ConvertPolicy policy, const ActivationLayerInfo &act_info)
Initialize the function's inputs and outputs.
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 of NEAddMulAdd.
Store the tensor's metadata.
#define ARM_COMPUTE_LOG_PARAMS(...)