24.02.1
|
Basic operator to run arm_compute::opencl::kernels::ClArithmeticKernel for PRELU. More...
#include <ClPRelu.h>
Public Member Functions | |
void | configure (const CLCompileContext &compile_context, ITensorInfo *input, ITensorInfo *alpha, ITensorInfo *output) |
Set the input and output tensor. More... | |
void | run (ITensorPack &tensors) override |
Run the kernels contained in the function. More... | |
Public Member Functions inherited from ICLOperator | |
ICLOperator (IRuntimeContext *ctx=nullptr) | |
Constructor. More... | |
ICLOperator (const ICLOperator &)=delete | |
Prevent instances of this class from being copied (As this class contains pointers) More... | |
ICLOperator (ICLOperator &&)=default | |
Default move constructor. More... | |
ICLOperator & | operator= (const ICLOperator &)=delete |
Prevent instances of this class from being copied (As this class contains pointers) More... | |
ICLOperator & | operator= (ICLOperator &&)=default |
Default move assignment operator. 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 *input, const ITensorInfo *alpha, const ITensorInfo *output) |
Static function to check if given info will lead to a valid configuration. More... | |
Basic operator to run arm_compute::opencl::kernels::ClArithmeticKernel for PRELU.
void configure | ( | const CLCompileContext & | compile_context, |
ITensorInfo * | input, | ||
ITensorInfo * | alpha, | ||
ITensorInfo * | output | ||
) |
Set the input and output tensor.
[in] | compile_context | The compile context to be used. |
[in] | input | Source tensor. Data types supported: QASYMM8/QASYMM8_SIGNED/F16/F32. |
[in] | alpha | PRelu layer parameters. Data types supported: same of input . |
[out] | output | Destination tensor. Data type supported: same as input |
Definition at line 34 of file ClPRelu.cpp.
References ARM_COMPUTE_LOG_PARAMS, arm_compute::test::validation::input, and arm_compute::PRELU.
|
overridevirtual |
Run the kernels contained in the function.
[in] | tensors | Vector that contains the tensors to operate on. |
Reimplemented from ICLOperator.
Definition at line 50 of file ClPRelu.cpp.
References arm_compute::ACL_DST, arm_compute::ACL_SRC_0, ITensorPack::add_tensor(), ARM_COMPUTE_ERROR_ON_MSG, ITensorPack::get_const_tensor(), ITensorPack::get_tensor(), and ICLOperator::run().
|
static |
Static function to check if given info will lead to a valid configuration.
Similar to ClPRelu::configure()
Definition at line 45 of file ClPRelu.cpp.
References arm_compute::test::validation::input, arm_compute::PRELU, and ClArithmeticKernel::validate().