diff options
Diffstat (limited to 'utils/CommonGraphOptions.h')
-rw-r--r-- | utils/CommonGraphOptions.h | 33 |
1 files changed, 17 insertions, 16 deletions
diff --git a/utils/CommonGraphOptions.h b/utils/CommonGraphOptions.h index 98f020d79e..c42e06cb84 100644 --- a/utils/CommonGraphOptions.h +++ b/utils/CommonGraphOptions.h @@ -24,13 +24,13 @@ #ifndef ARM_COMPUTE_EXAMPLES_UTILS_COMMON_GRAPH_OPTIONS #define ARM_COMPUTE_EXAMPLES_UTILS_COMMON_GRAPH_OPTIONS -#include "utils/command_line/CommandLineOptions.h" -#include "utils/command_line/CommandLineParser.h" - #include "arm_compute/graph/TypeLoader.h" #include "arm_compute/graph/TypePrinter.h" #include "arm_compute/runtime/CL/CLTunerTypes.h" +#include "utils/command_line/CommandLineOptions.h" +#include "utils/command_line/CommandLineParser.h" + namespace arm_compute { namespace utils @@ -40,7 +40,7 @@ namespace utils * * --help : Print the example's help message. * --threads : The number of threads to be used by the example during execution. - * --target : Execution target to be used by the examples. Supported target options: Neon, CL, GC. + * --target : Execution target to be used by the examples. Supported target options: Neon, CL, CLVK. * --type : Data type to be used by the examples. Supported data type options: QASYMM8, F16, F32. * --layout : Data layout to be used by the examples. Supported data layout options : NCHW, NHWC. * --enable-tuner : Toggle option to enable the OpenCL dynamic tuner. @@ -62,7 +62,6 @@ namespace utils * * Note that data, image and labels options should be provided to perform an inference run on an image. * Note that validation-file and validation-path should be provided to perform a graph accuracy estimation. - * Note GLES target is not supported for most of the networks. * * Example execution commands: * @@ -93,15 +92,16 @@ namespace utils /** Structure holding all the common graph parameters */ struct CommonGraphParams { - bool help{ false }; - int threads{ 0 }; - arm_compute::graph::Target target{ arm_compute::graph::Target::NEON }; - arm_compute::DataType data_type{ DataType::F32 }; - arm_compute::DataLayout data_layout{ DataLayout::NHWC }; - bool enable_tuner{ false }; - bool enable_cl_cache{ false }; - arm_compute::CLTunerMode tuner_mode{ CLTunerMode::NORMAL }; - arm_compute::graph::FastMathHint fast_math_hint{ arm_compute::graph::FastMathHint::Disabled }; + 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}; + bool enable_tuner{false}; + bool enable_cl_cache{false}; + arm_compute::CLTunerMode tuner_mode{CLTunerMode::NORMAL}; + arm_compute::graph::FastMathHint fast_math_hint{arm_compute::graph::FastMathHint::Disabled}; std::string data_path{}; std::string image{}; std::string labels{}; @@ -109,8 +109,8 @@ struct CommonGraphParams std::string validation_path{}; std::string tuner_file{}; std::string mlgo_file{}; - unsigned int validation_range_start{ 0 }; - unsigned int validation_range_end{ std::numeric_limits<unsigned int>::max() }; + unsigned int validation_range_start{0}; + unsigned int validation_range_end{std::numeric_limits<unsigned int>::max()}; }; /** Formatted output of the CommonGraphParams type @@ -152,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 */ |