24.02.1
|
Go to the documentation of this file.
24 #ifndef ARM_COMPUTE_CPU_ELEMENTWISE_H
25 #define ARM_COMPUTE_CPU_ELEMENTWISE_H
44 template <ArithmeticOperation op>
151 template <ComparisonOperation op>
void configure(const ITensorInfo *src0, const ITensorInfo *src1, ITensorInfo *dst)
Initialise the kernel's inputs, dst and conversion policy.
Basic function to run cpu::kernels::CpuComparisonKernel.
Basic function to run cpu::kernels::CpuArithmeticKernel for power.
static Status validate(const ITensorInfo *src0, const ITensorInfo *src1, const ITensorInfo *dst)
Static function to check if given info will lead to a valid configuration.
ComparisonOperation
Supported comparison operations.
Basic function to run cpu::kernels::CpuArithmeticKernel for division.
Basic function to run cpu::kernels::CpuComparisonKernel.
static Status validate(const ITensorInfo *src0, const ITensorInfo *src1, const ITensorInfo *dst)
Static function to check if given info will lead to a valid configuration.
void configure(const ITensorInfo *src0, const ITensorInfo *src1, ITensorInfo *dst)
Configure the operator.
static Status validate(const ITensorInfo *src0, const ITensorInfo *src1, const ITensorInfo *dst, ComparisonOperation op)
Static function to check if given info will lead to a valid configuration.
Basic interface for functions which have a single async CPU kernel.
static Status validate(const ITensorInfo *src0, const ITensorInfo *src1, const ITensorInfo *dst)
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.
void configure(const ITensorInfo *src0, const ITensorInfo *src1, ITensorInfo *dst)
Initialise the kernel's inputs, dst and conversion policy.
void configure(const ITensorInfo *src0, const ITensorInfo *src1, ITensorInfo *dst)
Initialise the kernel's inputs, dst and conversion policy.
Copyright (c) 2017-2024 Arm Limited.
Class to run cpu::kernels::CpuArithmeticKernel except for division and power.
void configure(const ITensorInfo *src0, const ITensorInfo *src1, ITensorInfo *dst, ComparisonOperation op)
Initialise the kernel's inputs, dst and conversion policy.
Store the tensor's metadata.
static Status validate(const ITensorInfo *src0, const ITensorInfo *src1, const ITensorInfo *dst)
Static function to check if given info will lead to a valid configuration.