diff options
Diffstat (limited to 'tests/tools/tflite_run/src/args.h')
| -rw-r--r-- | tests/tools/tflite_run/src/args.h | 6 |
1 files changed, 5 insertions, 1 deletions
diff --git a/tests/tools/tflite_run/src/args.h b/tests/tools/tflite_run/src/args.h index 5561544eb..25fd77a63 100644 --- a/tests/tools/tflite_run/src/args.h +++ b/tests/tools/tflite_run/src/args.h @@ -28,12 +28,14 @@ namespace TFLiteRun class Args { public: - Args(const int argc, char **argv); + Args(const int argc, char **argv) noexcept; void print(void); const std::string &getTFLiteFilename(void) const { return _tflite_filename; } const std::string &getDumpFilename(void) const { return _dump_filename; } const std::string &getCompareFilename(void) const { return _compare_filename; } + const std::string &getInputFilename(void) const { return _input_filename; } + const std::vector<int> &getInputShapes(void) const { return _input_shapes; } private: void Initialize(); @@ -46,6 +48,8 @@ private: std::string _tflite_filename; std::string _dump_filename; std::string _compare_filename; + std::string _input_filename; + std::vector<int> _input_shapes; }; } // end of namespace TFLiteRun |