24.02.1
|
Go to the documentation of this file.
24 #ifndef ARM_COMPUTE_GRAPH_DEPTH_TO_SPACE_LAYER_NODE_H
25 #define ARM_COMPUTE_GRAPH_DEPTH_TO_SPACE_LAYER_NODE_H
DepthToSpaceLayerNode(int block_shape)
Default Constructor.
NodeType type() const override
Returns node's type.
TensorDescriptor configure_output(size_t idx) const override
Calculates output configuration.
static TensorDescriptor compute_output_descriptor(const TensorDescriptor &input_descriptor, int block_shape)
Computes depth to space output descriptor.
int block_shape() const
Block shape policy accessor.
bool forward_descriptors() override
Forwards descriptor information to outputs if possible.
Copyright (c) 2017-2024 Arm Limited.
void accept(INodeVisitor &v) override
Accepts a node visitor.