1*c217d954SCole Faust /* 2*c217d954SCole Faust * Copyright (c) 2018-2019 Arm Limited. 3*c217d954SCole Faust * 4*c217d954SCole Faust * SPDX-License-Identifier: MIT 5*c217d954SCole Faust * 6*c217d954SCole Faust * Permission is hereby granted, free of charge, to any person obtaining a copy 7*c217d954SCole Faust * of this software and associated documentation files (the "Software"), to 8*c217d954SCole Faust * deal in the Software without restriction, including without limitation the 9*c217d954SCole Faust * rights to use, copy, modify, merge, publish, distribute, sublicense, and/or 10*c217d954SCole Faust * sell copies of the Software, and to permit persons to whom the Software is 11*c217d954SCole Faust * furnished to do so, subject to the following conditions: 12*c217d954SCole Faust * 13*c217d954SCole Faust * The above copyright notice and this permission notice shall be included in all 14*c217d954SCole Faust * copies or substantial portions of the Software. 15*c217d954SCole Faust * 16*c217d954SCole Faust * THE SOFTWARE IS PROVIDED "AS IS", WITHOUT WARRANTY OF ANY KIND, EXPRESS OR 17*c217d954SCole Faust * IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY, 18*c217d954SCole Faust * FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT. IN NO EVENT SHALL THE 19*c217d954SCole Faust * AUTHORS OR COPYRIGHT HOLDERS BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER 20*c217d954SCole Faust * LIABILITY, WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING FROM, 21*c217d954SCole Faust * OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN THE 22*c217d954SCole Faust * SOFTWARE. 23*c217d954SCole Faust */ 24*c217d954SCole Faust #ifndef ARM_COMPUTE_GRAPH_CHANNEL_SHUFFLE_LAYER_NODE_H 25*c217d954SCole Faust #define ARM_COMPUTE_GRAPH_CHANNEL_SHUFFLE_LAYER_NODE_H 26*c217d954SCole Faust 27*c217d954SCole Faust #include "arm_compute/graph/INode.h" 28*c217d954SCole Faust 29*c217d954SCole Faust namespace arm_compute 30*c217d954SCole Faust { 31*c217d954SCole Faust namespace graph 32*c217d954SCole Faust { 33*c217d954SCole Faust /** Channel Shuffle Layer node */ 34*c217d954SCole Faust class ChannelShuffleLayerNode final : public INode 35*c217d954SCole Faust { 36*c217d954SCole Faust public: 37*c217d954SCole Faust /** Constructor 38*c217d954SCole Faust * 39*c217d954SCole Faust * @param[in] num_groups Number of groups 40*c217d954SCole Faust */ 41*c217d954SCole Faust ChannelShuffleLayerNode(unsigned int num_groups); 42*c217d954SCole Faust /** Number of groups accessor 43*c217d954SCole Faust * 44*c217d954SCole Faust * @return Number of groups 45*c217d954SCole Faust */ 46*c217d954SCole Faust unsigned int num_groups() const; 47*c217d954SCole Faust 48*c217d954SCole Faust // Inherited overridden methods: 49*c217d954SCole Faust NodeType type() const override; 50*c217d954SCole Faust bool forward_descriptors() override; 51*c217d954SCole Faust TensorDescriptor configure_output(size_t idx) const override; 52*c217d954SCole Faust void accept(INodeVisitor &v) override; 53*c217d954SCole Faust 54*c217d954SCole Faust private: 55*c217d954SCole Faust unsigned int _num_groups; 56*c217d954SCole Faust }; 57*c217d954SCole Faust } // namespace graph 58*c217d954SCole Faust } // namespace arm_compute 59*c217d954SCole Faust #endif /* ARM_COMPUTE_GRAPH_CHANNEL_SHUFFLE_LAYER_NODE_H */ 60