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_