Browse Source

fix compile error

tags/v1.3.0
lichun 3 years ago
parent
commit
7d87e54235
1 changed files with 3 additions and 1 deletions
  1. +3
    -1
      inc/framework/generator/ge_generator.h

+ 3
- 1
inc/framework/generator/ge_generator.h View File

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


Loading…
Cancel
Save