24 #ifndef ARM_COMPUTE_GRAPH_CHANNEL_SHUFFLE_LAYER_NODE_H 25 #define ARM_COMPUTE_GRAPH_CHANNEL_SHUFFLE_LAYER_NODE_H 55 unsigned int _num_groups;
TensorDescriptor configure_output(size_t idx) const override
Calculates output configuration.
unsigned int num_groups() const
Number of groups accessor.
NodeType type() const override
Returns node's type.
Copyright (c) 2017-2021 Arm Limited.
ChannelShuffleLayerNode(unsigned int num_groups)
Constructor.
bool forward_descriptors() override
Forwards descriptor information to outputs if possible.
Channel Shuffle Layer node.
void accept(INodeVisitor &v) override
Accepts a node visitor.