24.02.1
|
Go to the documentation of this file.
76 output_desc.
shape.
set(idx_c,
src->desc().shape[idx_c]);
constexpr EdgeID EmptyEdgeID
SimpleTensor< float > src
ROIAlignLayerNode(ROIPoolingLayerInfo &pool_info)
Constructor.
unsigned int pooled_width() const
Get the pooled width of the layer.
TensorID input_id(size_t idx) const
Returns the tensor ID of a given input of the node.
NodeType type() const override
Returns node's type.
TensorDescriptor & desc()
TensorInfo metadata accessor.
Tensor * output(size_t idx) const
Returns the tensor of a given output of the node.
unsigned int pooled_height() const
Get the pooled height of the layer.
#define ARM_COMPUTE_ERROR_ON(cond)
If the condition is true then an error message is printed and an exception thrown.
void accept(INodeVisitor &v) override
Accepts a node visitor.
DataLayout layout
Data layout.
virtual void visit(INode &n)=0
Visit INode.
TensorShape shape
Tensor shape.
#define ARM_COMPUTE_UNUSED(...)
To avoid unused variables warnings.
bool forward_descriptors() override
Forwards descriptor information to outputs if possible.
size_t get_data_layout_dimension_index(const DataLayout &data_layout, const DataLayoutDimension &data_layout_dimension)
Get the index of the given dimension.
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.
ROI Pooling Layer Information class.
const ROIPoolingLayerInfo & pooling_info() const
ROIPoolingLayerInfo accessor.
TensorDescriptor configure_output(size_t idx) const override
Calculates output configuration.