Browse Source

Merge branch 'development' of https://gitee.com/wan_xuelei/graphengine into development

pull/134/head
wxl 5 years ago
parent
commit
23e1084b69
1 changed files with 4 additions and 5 deletions
  1. +4
    -5
      inc/framework/generator/ge_generator.h

+ 4
- 5
inc/framework/generator/ge_generator.h View File

@@ -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);


Loading…
Cancel
Save