diff options
Diffstat (limited to 'src/inference_engine_tflite_private.h')
-rw-r--r-- | src/inference_engine_tflite_private.h | 2 |
1 files changed, 2 insertions, 0 deletions
diff --git a/src/inference_engine_tflite_private.h b/src/inference_engine_tflite_private.h index 33dd1f4..d491500 100644 --- a/src/inference_engine_tflite_private.h +++ b/src/inference_engine_tflite_private.h @@ -88,6 +88,8 @@ namespace TFLiteImpl void FillLayerId(std::map<std::string, int>& layerId, std::map<std::string, inference_engine_tensor_info>& layers, const std::vector<int>& buffer); + int FillLayer(std::map<std::string, inference_engine_tensor_info>& layers, + std::map<std::string, int>& layerId); std::unique_ptr<tflite::Interpreter> mInterpreter; std::unique_ptr<tflite::FlatBufferModel> mFlatBuffModel; |