| @@ -77,13 +77,12 @@ class GeGenerator { | |||||
| /// @param [in] engine_type: specific engine. | /// @param [in] engine_type: specific engine. | ||||
| /// @param [out] model_buff: model buff of single op. | /// @param [out] model_buff: model buff of single op. | ||||
| /// @return SUCCESS or FAILED | /// @return SUCCESS or FAILED | ||||
| Status BuildSingleOpModel(OpDescPtr &op_desc, const vector<GeTensor> &inputs, | |||||
| const vector<GeTensor> &outputs, OpEngineType engine_type, | |||||
| ModelBufferData &model_buff); | |||||
| Status BuildSingleOpModel(OpDescPtr &op_desc, const vector<GeTensor> &inputs, const vector<GeTensor> &outputs, | |||||
| OpEngineType engine_type, ModelBufferData &model_buff); | |||||
| private: | private: | ||||
| Status GenerateModel(const Graph &graph, const string &file_name_prefix, | |||||
| const vector<GeTensor> &inputs, ge::ModelBufferData& model, bool is_offline = true); | |||||
| Status GenerateModel(const Graph &graph, const string &file_name_prefix, const vector<GeTensor> &inputs, | |||||
| 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); | ||||