From 32521430ad67172d372721979e8a32b7d4317620 Mon Sep 17 00:00:00 2001 From: Pablo Tello Date: Thu, 15 Nov 2018 14:43:10 +0000 Subject: COMPMID-1750: Add PriorBox operator to graph API. Change-Id: I5d2ed5dcc342abff8124762f7bdee587cdf20032 --- arm_compute/core/utils/misc/ShapeCalculator.h | 4 ++-- 1 file changed, 2 insertions(+), 2 deletions(-) (limited to 'arm_compute/core/utils') diff --git a/arm_compute/core/utils/misc/ShapeCalculator.h b/arm_compute/core/utils/misc/ShapeCalculator.h index 5c9457ed6b..da9ff56fd0 100644 --- a/arm_compute/core/utils/misc/ShapeCalculator.h +++ b/arm_compute/core/utils/misc/ShapeCalculator.h @@ -646,13 +646,13 @@ inline TensorShape compute_space_to_batch_shape(const ITensorInfo *input, const return output_shape; } + inline TensorShape compute_prior_box_shape(const ITensorInfo &input, const PriorBoxLayerInfo &info) { DataLayout data_layout = input.data_layout(); const size_t idx_w = get_data_layout_dimension_index(data_layout, DataLayoutDimension::WIDTH); const size_t idx_h = get_data_layout_dimension_index(data_layout, DataLayoutDimension::HEIGHT); - - const int num_priors = info.aspect_ratios().size() * info.min_sizes().size() + info.max_sizes().size(); + const int num_priors = info.aspect_ratios().size() * info.min_sizes().size() + info.max_sizes().size(); TensorShape output_shape{}; output_shape.set(0, input.dimension(idx_w) * input.dimension(idx_h) * num_priors * 4); -- cgit v1.2.1