24.02
|
Go to the documentation of this file.
34 struct NEPixelWiseMultiplication::Impl
36 const ITensor *src_0{
nullptr};
37 const ITensor *src_1{
nullptr};
38 ITensor *
dst{
nullptr};
39 std::unique_ptr<cpu::CpuMul> op{
nullptr};
66 _impl->src_0 = input1;
67 _impl->src_1 = input2;
69 _impl->op = std::make_unique<cpu::CpuMul>();
70 _impl->op->configure(input1->
info(), input2->
info(), output->
info(),
scale, overflow_policy, rounding_policy,
83 struct NEComplexPixelWiseMultiplication::Impl
86 ITensor *src_1{
nullptr};
87 ITensor *
dst{
nullptr};
88 std::unique_ptr<cpu::CpuComplexMul> op{
nullptr};
109 _impl->src_0 = input1;
110 _impl->src_1 = input2;
112 _impl->op = std::make_unique<cpu::CpuComplexMul>();
122 _impl->op->run(
pack);
~NEPixelWiseMultiplication()
Default Destructor.
void run() override
Run the kernels contained in the function.
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.
Interface for CPU tensor.
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.
void add_tensor(int id, ITensor *tensor)
Add tensor to the pack.
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...
~NEComplexPixelWiseMultiplication()
Default Destructor.
Activation Layer Information class.
virtual ITensorInfo * info() const =0
Interface to be implemented by the child class to return the tensor's metadata.
RoundingPolicy
Rounding method.
NEComplexPixelWiseMultiplication()
Default Constructor.
void configure(ITensor *input1, ITensor *input2, ITensor *output, const ActivationLayerInfo &act_info=ActivationLayerInfo())
Initialise the kernel's inputs, output.
NEPixelWiseMultiplication()
Default Constructor.
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...
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.
Copyright (c) 2017-2024 Arm Limited.
ConvertPolicy
Policy to handle integer overflow.
void run() override
Run the kernels contained in the function.
Store the tensor's metadata.