Browse Source

Add single_op model_id.

pull/1158/head
unknown 4 years ago
parent
commit
ff8c1179a8
1 changed files with 2 additions and 2 deletions
  1. +2
    -2
      ge/executor/ge_executor.cc

+ 2
- 2
ge/executor/ge_executor.cc View File

@@ -929,12 +929,12 @@ Status GeExecutor::GetMemAndWeightSize(const void *model_data, size_t model_size
}

Status GeExecutor::LoadSingleOp(const std::string &model_name, const ge::ModelData &modelData, void *stream,
SingleOp **single_op, const uint64_t model_id = 0) {
SingleOp **single_op, const uint64_t model_id) {
return SingleOpManager::GetInstance().GetOpFromModel(model_name, modelData, stream, single_op, model_id);
}

Status GeExecutor::LoadDynamicSingleOp(const std::string &model_name, const ge::ModelData &modelData, void *stream,
DynamicSingleOp **single_op, const uint64_t model_id = 0) {
DynamicSingleOp **single_op, const uint64_t model_id) {
return SingleOpManager::GetInstance().GetDynamicOpFromModel(model_name, modelData, stream, single_op, model_id);
}



Loading…
Cancel
Save