diff options
Diffstat (limited to 'src/graph/nodes/FlattenLayerNode.cpp')
| -rw-r--r-- | src/graph/nodes/FlattenLayerNode.cpp | 2 |
1 files changed, 1 insertions, 1 deletions
diff --git a/src/graph/nodes/FlattenLayerNode.cpp b/src/graph/nodes/FlattenLayerNode.cpp index 78b45dc305..baae555247 100644 --- a/src/graph/nodes/FlattenLayerNode.cpp +++ b/src/graph/nodes/FlattenLayerNode.cpp @@ -57,7 +57,7 @@ TensorDescriptor FlattenLayerNode::configure_output(size_t idx) const ARM_COMPUTE_ERROR_ON(src == nullptr); TensorDescriptor output_desc = src->desc(); - output_desc.shape.collapse(src->desc().shape.num_dimensions()); + output_desc.shape.collapse(3); return output_desc; } |