36 auto k = std::make_unique<NEAccumulateKernel>();
37 k->configure(input, output);
38 _kernel = std::move(k);
47 auto k = std::make_unique<NEAccumulateWeightedFP16Kernel>();
48 k->configure(input, alpha, output);
49 _kernel = std::move(k);
53 auto k = std::make_unique<NEAccumulateWeightedKernel>();
54 k->configure(input, alpha, output);
55 _kernel = std::move(k);
63 auto k = std::make_unique<NEAccumulateSquaredKernel>();
64 k->configure(input, shift, output);
65 _kernel = std::move(k);
~NEAccumulateWeighted()
Default destructor.
~NEAccumulateSquared()
Default destructor.
Interface for Neon tensor.
Copyright (c) 2017-2021 Arm Limited.
void configure(const ITensor *input, uint32_t shift, ITensor *output)
Set the input and accumulation tensors and the shift value.
void configure(const ITensor *input, ITensor *output)
Set the input and accumulation tensors.
~NEAccumulate()
Default destructor.
void configure(const ITensor *input, float alpha, ITensor *output, bool use_fp16=false)
Set the input and accumulation tensors, and the scale value.