fix compile fail bug

pull/1260/head
lichun 4 years ago
parent c86ae170b4
commit 25bb9e7af8

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