Browse Source

Fix ut.

tags/v1.3.0
zhaozhixuan 4 years ago
parent
commit
7ce31b2e0e
1 changed files with 4 additions and 3 deletions
  1. +4
    -3
      ge/hybrid/executor/subgraph_executor.cc

+ 4
- 3
ge/hybrid/executor/subgraph_executor.cc View File

@@ -103,9 +103,10 @@ Status SubgraphExecutor::InitInputsForUnknownShape(const std::vector<TensorValue
auto op_desc = input_node->GetOpDesc(); auto op_desc = input_node->GetOpDesc();
GE_CHECK_NOTNULL(op_desc); GE_CHECK_NOTNULL(op_desc);
auto output_desc = op_desc->MutableOutputDesc(kDataInputIndex); auto output_desc = op_desc->MutableOutputDesc(kDataInputIndex);
output_desc.SetShape(tensor_desc->GetShape());
output_desc.SetOriginShape(tensor_desc->GetOriginShape());
output_desc.SetDataType(tensor_desc->GetDataType());
GE_CHECK_NOTNULL(output_desc);
output_desc->SetShape(tensor_desc->GetShape());
output_desc->SetOriginShape(tensor_desc->GetOriginShape());
output_desc->SetDataType(tensor_desc->GetDataType());
} }
} }




Loading…
Cancel
Save