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