32 template <ElementWiseUnary op>
37 std::unique_ptr<OperatorType> cpu_op{
nullptr };
40 template <ElementWiseUnary op>
42 : _impl(
std::make_unique<Impl>())
45 template <ElementWiseUnary op>
47 template <ElementWiseUnary op>
49 template <ElementWiseUnary op>
52 template <ElementWiseUnary op>
57 _impl->cpu_op = std::make_unique<OperatorType>();
58 _impl->cpu_op->configure(op, *_impl->src->info(), *_impl->dst->info());
61 template <ElementWiseUnary op>
67 template <ElementWiseUnary op>
73 _impl->cpu_op->run(pack);
~NEElementwiseUnaryLayer()
Default Destructor.
void run() override
Run the kernels contained in the function.
Store the tensor's metadata.
Interface for Neon tensor.
SimpleTensor< float > src
Copyright (c) 2017-2021 Arm Limited.
static Status validate(ElementWiseUnary op, const ITensorInfo &src, const ITensorInfo &dst)
Static function to check if given info will lead to a valid configuration.
static Status validate(const ITensorInfo *input, const ITensorInfo *output)
Static function to check if given info will lead to a valid configuration.
void configure(const ITensor *input, ITensor *output)
Initialize the function.
NEElementwiseUnaryLayer & operator=(const NEElementwiseUnaryLayer &)=delete
Prevent instances of this class from being copied (As this class contains pointers) ...
NEElementwiseUnaryLayer()
Default Constructor.
Basic function to perform unary elementwise operations.
void add_tensor(int id, ITensor *tensor)
Add tensor to the pack.