| @@ -82,10 +82,17 @@ class GeGenerator { | |||||
| private: | private: | ||||
| Status GenerateModel(const Graph &graph, const string &file_name_prefix, const vector<GeTensor> &inputs, | Status GenerateModel(const Graph &graph, const string &file_name_prefix, const vector<GeTensor> &inputs, | ||||
| <<<<<<< HEAD | |||||
| ge::ModelBufferData& model, bool is_offline = true); | ge::ModelBufferData& model, bool is_offline = true); | ||||
| Status BuildSingleOp(OpDescPtr &op_desc, const vector<GeTensor> &inputs, const vector<GeTensor> &outputs, | Status BuildSingleOp(OpDescPtr &op_desc, const vector<GeTensor> &inputs, const vector<GeTensor> &outputs, | ||||
| const string &model_file_name, OpEngineType engine_type, | const string &model_file_name, OpEngineType engine_type, | ||||
| ModelBufferData &model_buff, bool is_offline = true); | ModelBufferData &model_buff, bool is_offline = true); | ||||
| ======= | |||||
| ge::ModelBufferData &model, bool is_offline = true); | |||||
| Status BuildSingleOp(OpDescPtr &op_desc, const vector<GeTensor> &inputs, const vector<GeTensor> &outputs, | |||||
| const string &model_file_name, OpEngineType engine_type, ModelBufferData &model_buff, | |||||
| bool is_offline = true); | |||||
| >>>>>>> Feature:single op support optional input | |||||
| 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); | ||||
| class Impl; | class Impl; | ||||