Class FlattenLayer

arm_compute/graph/frontend/Layers.h:701–715

Flatten Layer */

};
/** Flatten Layer */
class FlattenLayer final : public ILayer
{
public:
    /** Construct a flatten layer. */
    FlattenLayer()
    {
    }

    NodeID create_layer(IStream &s) override
    {
        NodeParams  common_params = {name(), s.hints().target_hint};
        NodeIdxPair input         = {s.tail_node(), 0};
        return GraphBuilder::add_flatten_node(s.graph(), common_params, input);
    }
};

/** Fully Connected Layer */
class FullyConnectedLayer final : public ILayer