Imported Upstream version 1.7.0

[platform/core/ml/nnfw.git] / tests / tools / tflite_run / src / args.h

diff --git a/tests/tools/tflite_run/src/args.h b/tests/tools/tflite_run/src/args.h
index 8ba13b8..054e47b 100644 (file)
--- a/tests/tools/tflite_run/src/args.h
+++ b/tests/tools/tflite_run/src/args.h
@@ -38,9 +38,12 @@ public:
   const std::vector<int> &getInputShapes(void) const { return _input_shapes; }
   const int getNumRuns(void) const { return _num_runs; }
   const int getWarmupRuns(void) const { return _warmup_runs; }
+  const int getRunDelay(void) const { return _run_delay; }
   const bool getGpuMemoryPoll(void) const { return _gpumem_poll; }
   const bool getMemoryPoll(void) const { return _mem_poll; }
   const bool getWriteReport(void) const { return _write_report; }
+  const bool getModelValidate(void) const { return _tflite_validate; }
+  const int getVerboseLevel(void) const { return _verbose_level; }
 
 private:
   void Initialize();
@@ -57,9 +60,12 @@ private:
   std::vector<int> _input_shapes;
   int _num_runs;
   int _warmup_runs;
+  int _run_delay;
   bool _gpumem_poll;
   bool _mem_poll;
   bool _write_report;
+  bool _tflite_validate;
+  int _verbose_level;
 };
 
 } // end of namespace TFLiteRun