24 #ifndef __ARM_COMPUTE_GRAPH_SPLIT_LAYER_NODE_H__ 25 #define __ARM_COMPUTE_GRAPH_SPLIT_LAYER_NODE_H__ 55 unsigned int num_splits,
unsigned int axis,
unsigned int idx);
65 unsigned int axis()
const;
75 unsigned int _num_splits;
NodeType type() const override
Returns node's type.
SplitLayerNode(unsigned int num_splits, unsigned int axis=0)
Default Constructor.
This file contains all available output stages for GEMMLowp on OpenCL.
unsigned int num_splits() const
Number of splits accessor.
void accept(INodeVisitor &v) override
Accepts a node visitor.
bool forward_descriptors() override
Forwards descriptor information to outputs if possible.
unsigned int axis() const
Split axis accessor.
Status validate() const override
Validate node.
static std::pair< TensorDescriptor, Coordinates > compute_output_descriptor(const TensorDescriptor &input_descriptor, unsigned int num_splits, unsigned int axis, unsigned int idx)
Computes split layer output descriptor.
TensorDescriptor configure_output(size_t idx) const override
Calculates output configuration.