!1262 fix compile fail bug

From: @lichun30
Reviewed-by: @xchu42,@wqtshg
Signed-off-by: @wqtshg
pull/1262/MERGE
mindspore-ci-bot 4 years ago committed by Gitee
commit 51901f654a

@ -29,9 +29,9 @@
#include "graph/op_desc.h"
#include "graph/detail/attributes_holder.h"
#include "omg/omg_inner_types.h"
#include "model/ge_root_model.h"
namespace ge {
class GeRootModel;
class GE_FUNC_VISIBILITY GeGenerator {
public:
static GeGenerator &GetInstance() {
@ -99,6 +99,8 @@ class GE_FUNC_VISIBILITY GeGenerator {
const string &model_file_name, OpEngineType engine_type, ModelBufferData &model_buff,
bool is_offline = true);
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);
class Impl;

Loading…
Cancel
Save