21.02
|
Basic function to run NEComplexPixelWiseMultiplicationKernel. More...
#include <NEPixelWiseMultiplication.h>
Public Member Functions | |
void | configure (ITensorInfo *input1, ITensorInfo *input2, ITensorInfo *output, const ActivationLayerInfo &act_info=ActivationLayerInfo()) |
Initialise the kernel's inputs, output. More... | |
Public Member Functions inherited from INEOperator | |
INEOperator (IRuntimeContext *ctx=nullptr) | |
Constructor. More... | |
INEOperator (const INEOperator &)=delete | |
Prevent instances of this class from being copied (As this class contains pointers) More... | |
INEOperator (INEOperator &&)=default | |
Default move constructor. More... | |
INEOperator & | operator= (const INEOperator &)=delete |
Prevent instances of this class from being copied (As this class contains pointers) More... | |
INEOperator & | operator= (INEOperator &&)=default |
Default move assignment operator. More... | |
~INEOperator () | |
Default destructor. More... | |
void | run (ITensorPack &tensors) override |
Run the kernels contained in the function. More... | |
void | prepare (ITensorPack &constants) override |
Prepare the function for executing. More... | |
MemoryRequirements | workspace () const override |
Return the memory requirements required by the workspace. More... | |
Public Member Functions inherited from IOperator | |
virtual | ~IOperator ()=default |
Destructor. More... | |
Static Public Member Functions | |
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 NEComplexPixelWiseMultiplication. More... | |
Basic function to run NEComplexPixelWiseMultiplicationKernel.
Definition at line 111 of file NEPixelWiseMultiplication.h.
void configure | ( | ITensorInfo * | input1, |
ITensorInfo * | input2, | ||
ITensorInfo * | output, | ||
const ActivationLayerInfo & | act_info = ActivationLayerInfo() |
||
) |
Initialise the kernel's inputs, output.
[in,out] | input1 | An input tensor. Data types supported: F32. Number of channels supported: 2 (complex tensor). The input tensor is [in, out] because its TensorInfo might be modified inside the kernel in case of broadcasting of dimension 0. |
[in,out] | input2 | An input tensor. Data types supported: same as input1 . Number of channels supported: same as input1 . The input tensor is [in, out] because its TensorInfo might be modified inside the kernel in case of broadcasting of dimension 0. |
[out] | output | The output tensor. Data types supported: same as input1 . Number of channels: same as input1 . |
[in] | act_info | (Optional) Activation layer information in case of a fused activation. Currently not supported. |
Definition at line 50 of file NEPixelWiseMultiplication.cpp.
References ARM_COMPUTE_UNUSED.
|
static |
Static function to check if given info will lead to a valid configuration of NEComplexPixelWiseMultiplication.
[in] | input1 | An input tensor info. Data types supported: F32. Number of channels supported: 2 (complex tensor). |
[in] | input2 | An input tensor info. Data types supported: same as input1 . Number of channels supported: same as input1 . |
[in] | output | The output tensor info. Data types supported: same as input1 . Number of channels supported: same as input1 . |
[in] | act_info | (Optional) Activation layer information in case of a fused activation. Currently not supported. |
Definition at line 58 of file NEPixelWiseMultiplication.cpp.
References ARM_COMPUTE_RETURN_ERROR_ON, arm_compute::test::validation::dst, ActivationLayerInfo::enabled(), and NEComplexPixelWiseMultiplicationKernel::validate().
Referenced by NEComplexPixelWiseMultiplication::validate().