19.08

Basic function to run NEPixelWiseMultiplicationKernel. More...
#include <NEPixelWiseMultiplication.h>
Public Member Functions  
void  configure (ITensor *input1, ITensor *input2, ITensor *output, float scale, ConvertPolicy overflow_policy, RoundingPolicy rounding_policy) 
Initialise the kernel's inputs, output and convertion policy. More...  
Public Member Functions inherited from INESimpleFunction  
INESimpleFunction ()  
Constructor. More...  
void  run () override final 
Run the kernels contained in the function. More...  
Public Member Functions inherited from IFunction  
virtual  ~IFunction ()=default 
Destructor. More...  
virtual void  prepare () 
Prepare the function for executing. More...  
Static Public Member Functions  
static Status  validate (const ITensorInfo *input1, const ITensorInfo *input2, const ITensorInfo *output, float scale, ConvertPolicy overflow_policy, RoundingPolicy rounding_policy) 
Static function to check if given info will lead to a valid configuration of NEPixelWiseMultiplication. More...  
Basic function to run NEPixelWiseMultiplicationKernel.
Definition at line 35 of file NEPixelWiseMultiplication.h.
void configure  (  ITensor *  input1, 
ITensor *  input2,  
ITensor *  output,  
float  scale,  
ConvertPolicy  overflow_policy,  
RoundingPolicy  rounding_policy  
) 
Initialise the kernel's inputs, output and convertion policy.
scale
equal to 1/255 only round to nearest even (implemented as round half up) is supported. For all other scale values only round to zero (implemented as round towards minus infinity) is supported.[in,out]  input1  An input tensor. Data types supported: U8/QASYMM8/S16/QSYMM16/F16/F32 This 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: U8, QASYMM8 (only if input1 is QASYMM8), S16, QSYMM16 (only if input1 is QSYMM16), F16 (only if input1 is F16), F32 (only if input1 is F32). This input tensor is [in, out] because its TensorInfo might be modified inside the kernel in case of broadcasting of dimension 0. 
[out]  output  Output tensor. Data types supported: U8 (Only if both inputs are U8), QASYMM8 (only if both inputs are QASYMM8), S16, QSYMM16 (only if both inputs are QSYMM16), F16 (only if input1 is F16), F32 (only if both inputs are F32). 
[in]  scale  Scale to apply after multiplication. Scale must be positive and its value must be either 1/255 or 1/2^n where n is between 0 and 15. 
[in]  overflow_policy  Overflow policy. ConvertPolicy cannot be WRAP if datatype is QASYMM8 or QSYMM16. 
[in]  rounding_policy  Rounding policy. 
Definition at line 34 of file NEPixelWiseMultiplication.cpp.
References ITensorInfo::dimension(), ITensor::info(), arm_compute::REPLICATE, arm_compute::test::validation::rounding_policy, and arm_compute::test::validation::scale.
Referenced by NELSTMLayerQuantized::configure().

static 
Static function to check if given info will lead to a valid configuration of NEPixelWiseMultiplication.
scale
equal to 1/255 only round to nearest even (implemented as round half up) is supported. For all other scale values only round to zero (implemented as round towards minus infinity) is supported.[in]  input1  An input tensor info. Data types supported: U8/QASYMM8/S16/QSYMM16/F16/F32 
[in]  input2  An input tensor info. Data types supported: U8, QASYMM8 (only if input1 is QASYMM8), S16, QSYMM16 (only if both inputs are QSYMM16), F16 (only if input1 is F16), F32 (only if input1 is F32). 
[in]  output  Output tensor info. Data types supported: U8 (Only if both inputs are U8), QASYMM8 (only if both inputs are QASYMM8), S16, QSYMM16 (only if both inputs are QSYMM16), F16 (only if input1 is F16), F32 (only if both inputs are F32). 
[in]  scale  Scale to apply after multiplication. Scale must be positive and its value must be either 1/255 or 1/2^n where n is between 0 and 15. 
[in]  overflow_policy  Overflow policy. ConvertPolicy cannot be WRAP if datatype is QASYMM8 or QSYMM16. 
[in]  rounding_policy  Rounding policy. 
Definition at line 50 of file NEPixelWiseMultiplication.cpp.
References arm_compute::test::validation::rounding_policy, arm_compute::test::validation::scale, and NEPixelWiseMultiplicationKernel::validate().
Referenced by NELSTMLayerQuantized::validate().