33 struct NEPixelWiseMultiplication::Impl
35 const ITensor *src_0{
nullptr };
36 const ITensor *src_1{
nullptr };
37 ITensor *
dst{
nullptr };
38 std::unique_ptr<cpu::CpuMul> op{
nullptr };
42 : _impl(
std::make_unique<Impl>())
50 return cpu::CpuMul::validate(input1, input2, output, scale, overflow_policy, rounding_policy, act_info);
56 _impl->src_0 = input1;
57 _impl->src_1 = input2;
59 _impl->op = std::make_unique<cpu::CpuMul>();
60 _impl->op->configure(input1->
info(), input2->
info(), output->
info(),
scale, overflow_policy, rounding_policy, act_info);
72 struct NEComplexPixelWiseMultiplication::Impl
77 std::unique_ptr<cpu::CpuComplexMul> op{
nullptr };
81 : _impl(
std::make_unique<Impl>())
93 _impl->src_0 = input1;
94 _impl->src_1 = input2;
96 _impl->op = std::make_unique<cpu::CpuComplexMul>();
97 _impl->op->configure(input1->
info(), input2->
info(), output->
info(), act_info);
106 _impl->op->run(pack);
~NEComplexPixelWiseMultiplication()
Default Destructor.
void run() override
Run the kernels contained in the function.
void run() override
Run the kernels contained in the function.
NEPixelWiseMultiplication()
Default Constructor.
Store the tensor's metadata.
static Status validate(const ITensorInfo *input1, const ITensorInfo *input2, const ITensorInfo *output, const ActivationLayerInfo &act_info=ActivationLayerInfo())
Static function to check if given info will lead to a valid configuration of NEComplexPixelWiseMultip...
Activation Layer Information class.
Interface for CPU tensor.
Copyright (c) 2017-2021 Arm Limited.
void configure(const ITensor *input1, const ITensor *input2, ITensor *output, float scale, ConvertPolicy overflow_policy, RoundingPolicy rounding_policy, const ActivationLayerInfo &act_info=ActivationLayerInfo())
Initialise the kernel's inputs, output and convertion policy.
RoundingPolicy
Rounding method.
virtual ITensorInfo * info() const =0
Interface to be implemented by the child class to return the tensor's metadata.
~NEPixelWiseMultiplication()
Default Destructor.
NEComplexPixelWiseMultiplication()
Default Constructor.
static Status validate(const ITensorInfo *src1, const ITensorInfo *src2, const ITensorInfo *dst, const ActivationLayerInfo &act_info=ActivationLayerInfo())
Static function to check if given info will lead to a valid configuration.
static Status validate(const ITensorInfo *input1, const ITensorInfo *input2, const ITensorInfo *output, float scale, ConvertPolicy overflow_policy, RoundingPolicy rounding_policy, const ActivationLayerInfo &act_info=ActivationLayerInfo())
Static function to check if given info will lead to a valid configuration of NEPixelWiseMultiplicatio...
static Status validate(const ITensorInfo *src1, const ITensorInfo *src2, const ITensorInfo *dst, float scale, ConvertPolicy overflow_policy, RoundingPolicy rounding_policy, const ActivationLayerInfo &act_info=ActivationLayerInfo())
Static function to check if given info will lead to a valid configuration.
void configure(ITensor *input1, ITensor *input2, ITensor *output, const ActivationLayerInfo &act_info=ActivationLayerInfo())
Initialise the kernel's inputs, output.
ConvertPolicy
Policy to handle overflow.
void add_tensor(int id, ITensor *tensor)
Add tensor to the pack.