diff options
Diffstat (limited to 'utils/CommonGraphOptions.h')
-rw-r--r-- | utils/CommonGraphOptions.h | 2 |
1 files changed, 2 insertions, 0 deletions
diff --git a/utils/CommonGraphOptions.h b/utils/CommonGraphOptions.h index f2cbd48b72..afdb78b1be 100644 --- a/utils/CommonGraphOptions.h +++ b/utils/CommonGraphOptions.h @@ -94,6 +94,7 @@ struct CommonGraphParams { 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 @@ public: 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 */ |