33 auto k = std::make_unique<NEChannelShuffleLayerKernel>();
34 k->configure(input, output, num_groups);
35 _kernel = std::move(k);
Store the tensor's metadata.
static Status validate(const ITensorInfo *input, const ITensorInfo *output, unsigned int num_groups)
Static function to check if given info will lead to a valid configuration of NEChannelShuffleLayerKer...
void configure(const ITensor *input, ITensor *output, unsigned int num_groups)
Initialize the function.
Interface for Neon tensor.
Copyright (c) 2017-2021 Arm Limited.
static Status validate(const ITensorInfo *input, const ITensorInfo *output, unsigned int num_groups)
Static function to check if given info will lead to a valid configuration of NEChannelShuffleLayer.
const unsigned int num_groups