diff options
Diffstat (limited to 'compiler/tflite2circle/include/TFLModel.h')
-rw-r--r-- | compiler/tflite2circle/include/TFLModel.h | 7 |
1 files changed, 3 insertions, 4 deletions
diff --git a/compiler/tflite2circle/include/TFLModel.h b/compiler/tflite2circle/include/TFLModel.h index e53d62749..507667bb9 100644 --- a/compiler/tflite2circle/include/TFLModel.h +++ b/compiler/tflite2circle/include/TFLModel.h @@ -37,15 +37,14 @@ public: TFLModel(const std::string &path); public: - bool is_valid(void) { return _valid; } + const tflite::Model *get_model(void); -private: - const tflite::Model *load_model(void); +public: + bool verify_data(void); private: std::ifstream _infile; DataBuffer _data; - bool _valid; friend class CircleModel; }; |