Browse Source

update opdesc

pull/1500/head
zhou_chao1993 4 years ago
parent
commit
361bbed5f8
1 changed files with 1 additions and 1 deletions
  1. +1
    -1
      ge/single_op/task/op_task.cc

+ 1
- 1
ge/single_op/task/op_task.cc View File

@@ -464,7 +464,7 @@ Status AiCpuBaseTask::UpdateExtInfo(const std::vector<GeTensorDesc> &input_desc,
GE_CHK_STATUS_RET(aicpu_ext_handle_->UpdateInputShapeAndType(input_index, input_desc[non_const_index]),
"Input[%zu] update input shape failed.", input_index);
if (DumpManager::GetInstance().GetDumpProperties(kInferSessionId).IsSingleOpNeedDump()) {
GE_CHK_STATUS_RET(op_desc_->UpdateInputDesc(input_index, input_desc[non_const_index]),
GE_CHK_STATUS_RET(op_desc_->UpdateInputDesc(input_index, input_desc[input_index]),
"AicpuTask Update [%zu]th input desc failed", input_index);
}
non_const_index++;


Loading…
Cancel
Save