24.02.1
|
Go to the documentation of this file.
56 output_descriptor.
shape =
59 return output_descriptor;
DepthToSpaceLayerNode(int block_shape)
Default Constructor.
constexpr EdgeID EmptyEdgeID
SimpleTensor< float > src
TensorShape compute_depth_to_space_shape(const TensorShape &input_shape, DataLayout data_layout, int block)
Calculate the depth to space output shape of a tensor.
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.
NodeType type() const override
Returns node's type.
constexpr auto data_layout
TensorDescriptor configure_output(size_t idx) const override
Calculates output configuration.
#define ARM_COMPUTE_ERROR_ON(cond)
If the condition is true then an error message is printed and an exception thrown.
static TensorDescriptor compute_output_descriptor(const TensorDescriptor &input_descriptor, int block_shape)
Computes depth to space output descriptor.
DataLayout layout
Data layout.
virtual void visit(INode &n)=0
Visit INode.
int block_shape() const
Block shape policy accessor.
TensorShape shape
Tensor shape.
#define ARM_COMPUTE_UNUSED(...)
To avoid unused variables warnings.
TensorShape input_shape
Validate test suite is to test ARM_COMPUTE_RETURN_ON_* macros we use to check the validity of given a...
bool forward_descriptors() override
Forwards descriptor information to outputs if possible.
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.
void accept(INodeVisitor &v) override
Accepts a node visitor.