aboutsummaryrefslogtreecommitdiff
path: root/tests/validation/Helpers.h
diff options
context:
space:
mode:
Diffstat (limited to 'tests/validation/Helpers.h')
-rw-r--r--tests/validation/Helpers.h14
1 files changed, 14 insertions, 0 deletions
diff --git a/tests/validation/Helpers.h b/tests/validation/Helpers.h
index c0c5865e56..d92699d93e 100644
--- a/tests/validation/Helpers.h
+++ b/tests/validation/Helpers.h
@@ -27,8 +27,12 @@
#include "Types.h"
#include "ValidationUserConfiguration.h"
+#include "arm_compute/core/Types.h"
+
+#include <random>
#include <type_traits>
#include <utility>
+#include <vector>
namespace arm_compute
{
@@ -168,6 +172,16 @@ inline void fill_mask_from_pattern(uint8_t *mask, int cols, int rows, MatrixPatt
}
}
+/** Create a vector of random ROIs.
+ *
+ * @param[in] shape The shape of the input tensor.
+ * @param[in] pool_info The ROI pooling information.
+ * @param[in] num_rois The number of ROIs to be created.
+ * @param[in] seed The random seed to be used.
+ *
+ * @return A vector that contains the requested number of random ROIs
+ */
+std::vector<ROI> generate_random_rois(const TensorShape &shape, const ROIPoolingLayerInfo &pool_info, unsigned int num_rois, std::random_device::result_type seed);
} // namespace validation
} // namespace test
} // namespace arm_compute