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