24.02.1
|
Go to the documentation of this file.
54 "The height of the input tensor must be a multiple of stride");
63 return output_descriptor;
int stride() const
Stride value to use for reorganizing the values in the output tensor.
constexpr EdgeID EmptyEdgeID
SimpleTensor< float > src
bool forward_descriptors() override
Forwards descriptor information to outputs if possible.
DataLayout
[DataLayout enum definition]
TensorID input_id(size_t idx) const
Returns the tensor ID of a given input of the node.
Tensor * output(size_t idx) const
Returns the tensor of a given output of the node.
size_t get_dimension_size(const TensorDescriptor &descriptor, const DataLayoutDimension data_layout_dimension)
Get size of a tensor's given dimension depending on its layout.
ReorgLayerNode(int stride)
Constructor.
constexpr auto data_layout
#define ARM_COMPUTE_ERROR_ON(cond)
If the condition is true then an error message is printed and an exception thrown.
#define ARM_COMPUTE_ERROR_ON_MSG(cond, msg)
DataLayout layout
Data layout.
virtual void visit(INode &n)=0
Visit INode.
TensorDescriptor configure_output(size_t idx) const override
Calculates output configuration.
static TensorDescriptor compute_output_descriptor(const TensorDescriptor &input_descriptor, int stride)
Computes reorg output descriptor.
NodeType type() const override
Returns node's type.
TensorShape shape
Tensor shape.
#define ARM_COMPUTE_UNUSED(...)
To avoid unused variables warnings.
size_t get_dimension_idx(DataLayout data_layout, const DataLayoutDimension data_layout_dimension)
Get index of a tensor's given dimension depending on its layout.
void accept(INodeVisitor &v) override
Accepts a node visitor.
Copyright (c) 2017-2024 Arm Limited.
Tensor * input(size_t idx) const
Returns the tensor of a given input of the node.
TensorID output_id(size_t idx) const
Returns the tensor ID of a given output of the node.
constexpr TensorID NullTensorID
Constant NodeID specifying an equivalent of null node.
TensorShape & set(size_t dimension, size_t value, bool apply_dim_correction=true, bool increase_dim_unit=true)
Accessor to set the value of one of the dimensions.