diff options
author | Chunseok Lee <chunseok.lee@samsung.com> | 2019-01-08 17:36:34 +0900 |
---|---|---|
committer | Chunseok Lee <chunseok.lee@samsung.com> | 2019-01-08 17:36:34 +0900 |
commit | bd11b24234d7d43dfe05a81c520aa01ffad06e42 (patch) | |
tree | 57d0d4044977e4fa0e50cd9ba40b32006dff19eb /tools/tflite_run/src/tensor_loader.h | |
parent | 91f4ba45449f700a047a4aeea00b1a7c84e94c75 (diff) | |
download | nnfw-bd11b24234d7d43dfe05a81c520aa01ffad06e42.tar.gz nnfw-bd11b24234d7d43dfe05a81c520aa01ffad06e42.tar.bz2 nnfw-bd11b24234d7d43dfe05a81c520aa01ffad06e42.zip |
Imported Upstream version 0.3upstream/0.3
Diffstat (limited to 'tools/tflite_run/src/tensor_loader.h')
-rw-r--r-- | tools/tflite_run/src/tensor_loader.h | 35 |
1 files changed, 0 insertions, 35 deletions
diff --git a/tools/tflite_run/src/tensor_loader.h b/tools/tflite_run/src/tensor_loader.h deleted file mode 100644 index f2a699185..000000000 --- a/tools/tflite_run/src/tensor_loader.h +++ /dev/null @@ -1,35 +0,0 @@ -#ifndef __TFLITE_RUN_TENSOR_LOADER_H__ -#define __TFLITE_RUN_TENSOR_LOADER_H__ - -#include <sys/mman.h> - -#include <string> -#include <unordered_map> - -#include "support/tflite/TensorView.h" - -namespace tflite -{ -class Interpreter; -} - -namespace TFLiteRun -{ - -class TensorLoader -{ -public: - TensorLoader(tflite::Interpreter &interpreter); - void load(const std::string &filename); - const nnfw::support::tflite::TensorView<float> &get(int tensor_idx) const; - size_t getNums() const { return _tensor_map.size(); } - -private: - tflite::Interpreter &_interpreter; - std::unique_ptr<float> _raw_data; - std::unordered_map<int, nnfw::support::tflite::TensorView<float>> _tensor_map; -}; - -} // end of namespace TFLiteRun - -#endif // __TFLITE_RUN_TENSOR_LOADER_H__ |