diff options
| author | Pablo Tello <pablo.tello@arm.com> | 2018-11-15 14:43:10 +0000 |
|---|---|---|
| committer | Anthony Barbier <Anthony.barbier@arm.com> | 2018-11-19 09:58:20 +0000 |
| commit | 32521430ad67172d372721979e8a32b7d4317620 (patch) | |
| tree | c18921290341c1d730fbff406960dc7ea7f61306 /src/graph/nodes/FlattenLayerNode.cpp | |
| parent | d696cb6d18c2fe66f1abce88bbd14faf2137ef89 (diff) | |
| download | ComputeLibrary-32521430ad67172d372721979e8a32b7d4317620.tar.gz | |
COMPMID-1750: Add PriorBox operator to graph API.
Change-Id: I5d2ed5dcc342abff8124762f7bdee587cdf20032
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; } |