23.11
|
Go to the documentation of this file.
37 if (_kernel->is_window_configured())
45 auto shape_and_window = compute_output_shape_and_window(src0_info->tensor_shape(), src1_info->tensor_shape());
49 template <ArithmeticOperation op>
53 auto k = std::make_unique<kernels::CpuArithmeticKernel>();
54 k->configure(op, src0, src1,
dst);
55 _kernel = std::move(k);
58 template <ArithmeticOperation op>
72 auto k = std::make_unique<kernels::CpuDivisionKernel>();
73 k->configure(src0, src1,
dst);
74 _kernel = std::move(k);
85 auto k = std::make_unique<kernels::CpuPowerKernel>();
86 k->configure(src0, src1,
dst);
87 _kernel = std::move(k);
95 template <ComparisonOperation COP>
99 auto k = std::make_unique<kernels::CpuComparisonKernel>();
100 k->configure(COP, src0, src1,
dst);
101 _kernel = std::move(k);
104 template <ComparisonOperation COP>
117 auto k = std::make_unique<kernels::CpuComparisonKernel>();
118 k->configure(op, src0, src1,
dst);
119 _kernel = std::move(k);
void configure(const ITensorInfo *src0, const ITensorInfo *src1, ITensorInfo *dst)
Initialise the kernel's inputs, dst and conversion policy.
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.
Basic function to run cpu::kernels::CpuComparisonKernel.
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.
virtual ITensorInfo * info() const =0
Interface to be implemented by the child class to return the tensor's metadata.
const ITensor * get_const_tensor(int id) const
Get constant tensor of a given id.
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.
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.
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)
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.
void run(ITensorPack &tensors) override
Run the kernels contained in the function.
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-2023 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.
#define ARM_COMPUTE_LOG_PARAMS(...)