33 auto k = std::make_unique<kernels::CpuArithmeticKernel>();
35 _kernel = std::move(k);
45 auto k = std::make_unique<kernels::CpuArithmeticKernel>();
47 _kernel = std::move(k);
57 auto k = std::make_unique<kernels::CpuArithmeticKernel>();
59 _kernel = std::move(k);
69 auto k = std::make_unique<kernels::CpuDivisionKernel>();
70 k->configure(input1, input2, output);
71 _kernel = std::move(k);
81 auto k = std::make_unique<kernels::CpuPowerKernel>();
82 k->configure(input1, input2, output);
83 _kernel = std::move(k);
91 template <ComparisonOperation COP>
94 auto k = std::make_unique<kernels::CpuComparisonKernel>();
95 k->configure(COP, input1, input2, output);
96 _kernel = std::move(k);
99 template <ComparisonOperation COP>
107 auto k = std::make_unique<kernels::CpuComparisonKernel>();
108 k->configure(op, input1, input2, output);
109 _kernel = std::move(k);
static Status validate(const ITensorInfo *input1, const ITensorInfo *input2, const ITensorInfo *output)
Static function to check if given info will lead to a valid configuration of CpuPowerKernel.
static Status validate(const ITensorInfo *input1, const ITensorInfo *input2, const ITensorInfo *output)
Static function to check if given info will lead to a valid configuration of cpu::kernels::CpuCompari...
void configure(const ITensorInfo *input1, const ITensorInfo *input2, ITensorInfo *output)
Initialise the kernel's inputs, output and conversion policy.
void configure(const ITensorInfo *input1, const ITensorInfo *input2, ITensorInfo *output)
Initialise the kernel's inputs, output and conversion policy.
Store the tensor's metadata.
void configure(const ITensorInfo *input1, const ITensorInfo *input2, ITensorInfo *output)
Initialise the kernel's inputs, output and conversion policy.
Copyright (c) 2017-2021 Arm Limited.
static Status validate(const ITensorInfo *input1, const ITensorInfo *input2, const ITensorInfo *output)
Static function to check if given info will lead to a valid configuration of cpu::kernels::CpuArithme...
ComparisonOperation
Supported comparison operations.
static Status validate(const ITensorInfo *input1, const ITensorInfo *input2, const ITensorInfo *output)
Static function to check if given info will lead to a valid configuration of cpu::kernels::CpuArithme...
void configure(const ITensorInfo *input1, const ITensorInfo *input2, ITensorInfo *output, ComparisonOperation op)
Initialise the kernel's inputs, output and conversion policy.
Basic function to run cpu::kernels::CpuComparisonKernel.
static Status validate(ComparisonOperation op, const ITensorInfo *input1, const ITensorInfo *input2, const ITensorInfo *output)
Static function to check if given info will lead to a valid configuration of cpu::kernels::CpuCompari...
static Status validate(const ITensorInfo *input1, const ITensorInfo *input2, const ITensorInfo *output)
Static function to check if given info will lead to a valid configuration of cpu::kernels::CpuArithme...
static Status validate(const ITensorInfo *input1, const ITensorInfo *input2, const ITensorInfo *output)
Static function to check if given info will lead to a valid configuration of cpu::kernels::CpuArithme...
void configure(const ITensorInfo *input1, const ITensorInfo *input2, ITensorInfo *output)
Initialise the kernel's inputs, output and conversion policy.
static Status validate(ArithmeticOperation op, const ITensorInfo *input1, const ITensorInfo *input2, const ITensorInfo *output)
Static function to check if given info will lead to a valid configuration of cpu::kernels::CpuArithme...
static Status validate(const ITensorInfo *input1, const ITensorInfo *input2, const ITensorInfo *output, ComparisonOperation op)
Static function to check if given info will lead to a valid configuration of cpu::kernels::CpuCompari...
static Status validate(const ITensorInfo *input1, const ITensorInfo *input2, const ITensorInfo *output)
Static function to check if given info will lead to a valid configuration of cpu::kernels::CpuArithme...
static Status validate(const ITensorInfo *input1, const ITensorInfo *input2, const ITensorInfo *output)
Static function to check if given info will lead to a valid configuration of CpuDivisionKernel.
void configure(const ITensorInfo *input1, const ITensorInfo *input2, ITensorInfo *output)
Initialise the kernel's inputs, output and conversion policy.
void configure(const ITensorInfo *input1, const ITensorInfo *input2, ITensorInfo *output)
Initialise the kernel's inputs, output and conversion policy.