Browse Source

Feature:single op support optional input

pull/137/head
wxl 5 years ago
parent
commit
37c582409a
1 changed files with 7 additions and 0 deletions
  1. +7
    -0
      inc/framework/generator/ge_generator.h

+ 7
- 0
inc/framework/generator/ge_generator.h View File

@@ -82,10 +82,17 @@ class GeGenerator {

private:
Status GenerateModel(const Graph &graph, const string &file_name_prefix, const vector<GeTensor> &inputs,
<<<<<<< HEAD
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);
=======
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);

class Impl;


Loading…
Cancel
Save