Add `batches` options in the graph examples
Signed-off-by: Georgios Pinitas <georgios.pinitas@arm.com>
Change-Id: Ib0442a25cb60793f530e1cff5e2aaf167d4a6950
Reviewed-on: https://review.mlplatform.org/c/ml/ComputeLibrary/+/5812
Reviewed-by: Michalis Spyrou <michalis.spyrou@arm.com>
Comments-Addressed: Arm Jenkins <bsgcomp@arm.com>
Tested-by: Arm Jenkins <bsgcomp@arm.com>
diff --git a/utils/CommonGraphOptions.h b/utils/CommonGraphOptions.h
index f2cbd48..afdb78b 100644
--- a/utils/CommonGraphOptions.h
+++ b/utils/CommonGraphOptions.h
@@ -94,6 +94,7 @@
{
bool help{ false };
int threads{ 0 };
+ int batches{ 1 };
arm_compute::graph::Target target{ arm_compute::graph::Target::NEON };
arm_compute::DataType data_type{ DataType::F32 };
arm_compute::DataLayout data_layout{ DataLayout::NHWC };
@@ -151,6 +152,7 @@
ToggleOption *help; /**< Show help option */
SimpleOption<int> *threads; /**< Number of threads option */
+ SimpleOption<int> *batches; /**< Number of batches */
EnumOption<arm_compute::graph::Target> *target; /**< Graph execution target */
EnumOption<arm_compute::DataType> *data_type; /**< Graph data type */
EnumOption<arm_compute::DataLayout> *data_layout; /**< Graph data layout */