Searched refs:NEChannelShuffleLayerKernel (Results 1 – 5 of 5) sorted by relevance
35 class NEChannelShuffleLayerKernel : public INEKernel43 NEChannelShuffleLayerKernel();45 NEChannelShuffleLayerKernel(const NEChannelShuffleLayerKernel &) = delete;47 NEChannelShuffleLayerKernel &operator=(const NEChannelShuffleLayerKernel &) = delete;49 NEChannelShuffleLayerKernel(NEChannelShuffleLayerKernel &&) = default;51 NEChannelShuffleLayerKernel &operator=(NEChannelShuffleLayerKernel &&) = default;53 ~NEChannelShuffleLayerKernel() = default;
136 NEChannelShuffleLayerKernel::NEChannelShuffleLayerKernel() in NEChannelShuffleLayerKernel() function in arm_compute::NEChannelShuffleLayerKernel141 void NEChannelShuffleLayerKernel::configure(const ITensor *input, ITensor *output, unsigned int num… in configure()165 Status NEChannelShuffleLayerKernel::validate(const ITensorInfo *input, const ITensorInfo *output, u… in validate()171 void NEChannelShuffleLayerKernel::run(const Window &window, const ThreadInfo &info) in run()
34 auto k = arm_compute::support::cpp14::make_unique<NEChannelShuffleLayerKernel>(); in configure()41 return NEChannelShuffleLayerKernel::validate(input, output, num_groups); in validate()
520 "src/core/NEON/kernels/NEChannelShuffleLayerKernel.cpp",
878 - @ref NEChannelShuffleLayer / @ref NEChannelShuffleLayerKernel