|
|
|
@@ -204,6 +204,8 @@ Status KnownNodeExecutor::LoadTask(const HybridModel &model, const NodePtr &node |
|
|
|
davinci_model->SetId(model.GetModelId()); |
|
|
|
davinci_model->SetDumpModelName(model.GetModelName()); |
|
|
|
davinci_model->SetOmName(model.GetOmName()); |
|
|
|
TensorValue *global_step_var = model.GetVariable(NODE_NAME_GLOBAL_STEP); |
|
|
|
davinci_model->SetGlobalStep(global_step_var->MutableData()); |
|
|
|
// set model id as root node's node id |
|
|
|
davinci_model->SetSubModelId(node->GetOpDesc()->GetId()); |
|
|
|
GELOGD("KnownNodeExecutor::LoadTask node id %ld.", node->GetOpDesc()->GetId()); |
|
|
|
|