| @@ -29,9 +29,9 @@ | |||||
| #include "graph/op_desc.h" | #include "graph/op_desc.h" | ||||
| #include "graph/detail/attributes_holder.h" | #include "graph/detail/attributes_holder.h" | ||||
| #include "omg/omg_inner_types.h" | #include "omg/omg_inner_types.h" | ||||
| #include "model/ge_root_model.h" | |||||
| namespace ge { | namespace ge { | ||||
| class GeRootModel; | |||||
| class GE_FUNC_VISIBILITY GeGenerator { | class GE_FUNC_VISIBILITY GeGenerator { | ||||
| public: | public: | ||||
| static GeGenerator &GetInstance() { | static GeGenerator &GetInstance() { | ||||
| @@ -99,6 +99,8 @@ class GE_FUNC_VISIBILITY GeGenerator { | |||||
| const string &model_file_name, OpEngineType engine_type, ModelBufferData &model_buff, | const string &model_file_name, OpEngineType engine_type, ModelBufferData &model_buff, | ||||
| bool is_offline = true); | bool is_offline = true); | ||||
| Status CheckForSingleOp(OpDescPtr &op_desc, const vector<GeTensor> &inputs, const vector<GeTensor> &outputs); | Status CheckForSingleOp(OpDescPtr &op_desc, const vector<GeTensor> &inputs, const vector<GeTensor> &outputs); | ||||
| using GeRootModelPtr = std::shared_ptr<ge::GeRootModel>; | |||||
| Status SetModelNameForDump(const GeRootModelPtr &ge_root_model); | Status SetModelNameForDump(const GeRootModelPtr &ge_root_model); | ||||
| class Impl; | class Impl; | ||||