Line data    Source code

       1              : // SPDX-License-Identifier: Apache-2.0
       2              : /**
       3              :  * Copyright (C) 2021 Jihoon Lee <jhoon.it.lee@samsung.com>
       4              :  *
       5              :  * @file tflite_interpreter.h
       6              :  * @date 12 April 2021
       7              :  * @brief NNTrainer *.tflite Interpreter
       8              :  * @see https://github.com/nnstreamer/nntrainer
       9              :  * @author Jihoon Lee <jhoon.it.lee@samsung.com>
      10              :  * @bug No known bugs except for NYI items
      11              :  */
      12              : #ifndef __TFLITE_INTERPRETER_H__
      13              : #define __TFLITE_INTERPRETER_H__
      14              : 
      15              : #include <interpreter.h>
      16              : 
      17              : #include <engine.h>
      18              : namespace nntrainer {
      19              : 
      20              : /**
      21              :  * @brief tflite graph interpreter class
      22              :  *
      23              :  */
      24              : class TfliteInterpreter : public GraphInterpreter {
      25              : public:
      26              :   /**
      27              :    * @brief Construct a new tflite Graph Interpreter object
      28              :    *
      29              :    */
      30              :   TfliteInterpreter() = default;
      31              : 
      32              :   /**
      33              :    * @brief Destroy the Tflite Interpreter object
      34              :    *
      35              :    */
      36            5 :   virtual ~TfliteInterpreter() = default;
      37              : 
      38              :   /**
      39              :    * @copydoc GraphInterpreter::serialize(const std::string &out)
      40              :    */
      41              :   void serialize(const GraphRepresentation &representation,
      42              :                  const std::string &out) override;
      43              : 
      44              :   /**
      45              :    * @copydoc GraphInterpreter::deserialize(const std::string &in)
      46              :    */
      47              :   GraphRepresentation deserialize(const std::string &in) override;
      48              : };
      49              : 
      50              : } // namespace nntrainer
      51              : 
      52              : #endif // __TFLITE_INTERPRETER_H__