24.02.1
|
Go to the documentation of this file.
24 #ifndef ARM_COMPUTE_GRAPH_SPLIT_LAYER_NODE_H
25 #define ARM_COMPUTE_GRAPH_SPLIT_LAYER_NODE_H
70 unsigned int axis()
const;
80 unsigned int _num_splits;
82 std::vector<int> _size_splits;
void accept(INodeVisitor &v) override
Accepts a node visitor.
unsigned int num_splits() const
Number of splits accessor.
SplitLayerNode(unsigned int num_splits, int axis=0, std::vector< int > size_splits=std::vector< int >())
Default Constructor.
bool forward_descriptors() override
Forwards descriptor information to outputs if possible.
TensorDescriptor configure_output(size_t idx) const override
Calculates output configuration.
NodeType type() const override
Returns node's type.
unsigned int axis() const
Split axis accessor.
Status validate() const override
Validate node.
Copyright (c) 2017-2024 Arm Limited.
std::pair< TensorDescriptor, Coordinates > compute_output_descriptor(const TensorDescriptor &input_descriptor, unsigned int num_splits, int axis, unsigned int idx)
Computes split layer output descriptor.