36 if(_kernel->is_window_configured())
44 auto shape_and_window = compute_output_shape_and_window(src0_info->tensor_shape(), src1_info->tensor_shape());
48 template <ArithmeticOperation op>
52 auto k = std::make_unique<kernels::CpuArithmeticKernel>();
53 k->configure(op, src0, src1, dst);
54 _kernel = std::move(
k);
57 template <ArithmeticOperation op>
71 auto k = std::make_unique<kernels::CpuDivisionKernel>();
72 k->configure(src0, src1, dst);
73 _kernel = std::move(
k);
84 auto k = std::make_unique<kernels::CpuPowerKernel>();
85 k->configure(src0, src1, dst);
86 _kernel = std::move(
k);
94 template <ComparisonOperation COP>
98 auto k = std::make_unique<kernels::CpuComparisonKernel>();
99 k->configure(COP, src0, src1, dst);
100 _kernel = std::move(
k);
103 template <ComparisonOperation COP>
112 auto k = std::make_unique<kernels::CpuComparisonKernel>();
113 k->configure(op, src0, src1, dst);
114 _kernel = std::move(
k);
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.
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.
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.
void configure(const ITensorInfo *src0, const ITensorInfo *src1, ITensorInfo *dst)
Initialise the kernel's inputs, dst and conversion policy.
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.
Store the tensor's metadata.
Copyright (c) 2017-2022 Arm Limited.
static Status validate(ArithmeticOperation op, const ITensorInfo *src0, const ITensorInfo *src1, const ITensorInfo *dst)
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.
void run(ITensorPack &tensors) override
Run the kernels contained in the function.
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.
virtual ITensorInfo * info() const =0
Interface to be implemented by the child class to return the tensor's metadata.
ComparisonOperation
Supported comparison operations.
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.
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.
Class to run cpu::kernels::CpuArithmeticKernel except for division and power.
void run(ITensorPack &tensors) override
Run the kernels contained in the function.
void configure(const ITensorInfo *src0, const ITensorInfo *src1, ITensorInfo *dst, ComparisonOperation op)
Initialise the kernel's inputs, dst and conversion policy.
#define ARM_COMPUTE_LOG_PARAMS(...)
void configure(const ITensorInfo *src0, const ITensorInfo *src1, ITensorInfo *dst)
Configure the operator.
static Status validate(ComparisonOperation op, const ITensorInfo *src0, const ITensorInfo *src1, const ITensorInfo *dst)
Static function to check if given info will lead to a valid configuration.
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.