Program Listing for File model_parser.h

Return to documentation for file (include/model_parser.h)

#ifndef MINDSPORE_LITE_INCLUDE_REGISTRY_MODEL_PARSER_H_
#define MINDSPORE_LITE_INCLUDE_REGISTRY_MODEL_PARSER_H_

#include "api/ir/func_graph.h"
#include "include/registry/converter_context.h"

namespace mindspore {
namespace converter {
class  ModelParser {
 public:
  ModelParser() = default;

  virtual ~ModelParser() = default;

  virtual api::FuncGraphPtr Parse(const converter::ConverterParameters &flags) { return this->res_graph_; }

 protected:
  api::FuncGraphPtr res_graph_ = nullptr;
};
}  // namespace converter
}  // namespace mindspore

#endif  // MINDSPORE_LITE_INCLUDE_REGISTRY_MODEL_PARSER_H_