24.02.1
|
Go to the documentation of this file.
40 auto k = std::make_unique<KernelType>();
42 _kernel = std::move(k);
static Status validate(ArithmeticOperation op, 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 run(ITensorPack &tensors) override
Run the kernels contained in the function.
@ PRELU
y*x if x < 0, x otherwise
void run(ITensorPack &tensors) override
Run the kernels contained in the function.
Interface for CPU tensor.
void add_tensor(int id, ITensor *tensor)
Add tensor to the pack.
ITensor * get_tensor(int id)
Get tensor of a given id from the pac.
void configure(const CLCompileContext &compile_context, ITensorInfo *input, ITensorInfo *alpha, ITensorInfo *output)
Set the input and output tensor.
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.
const ITensor * get_const_tensor(int id) const
Get constant tensor of a given id.
#define ARM_COMPUTE_ERROR_ON_MSG(cond, msg)
Copyright (c) 2017-2024 Arm Limited.
Store the tensor's metadata.
#define ARM_COMPUTE_LOG_PARAMS(...)