summaryrefslogtreecommitdiff
path: root/compiler/tflite2circle/include/TFLModel.h
diff options
context:
space:
mode:
Diffstat (limited to 'compiler/tflite2circle/include/TFLModel.h')
-rw-r--r--compiler/tflite2circle/include/TFLModel.h7
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;
};