diff options
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__ |