|
|
@@ -721,8 +721,12 @@ Status GeGenerator::BuildSingleOp(OpDescPtr &op_desc, const vector<GeTensor> &in |
|
|
|
GeModelPtr &ge_model = name_to_ge_model.begin()->second; |
|
|
|
GELOGD("The opType in op_desc_tmp is [%s]", op_desc_tmp->GetType().c_str()); |
|
|
|
|
|
|
|
bool all_shape = false; |
|
|
|
(void)AttrUtils::GetBool(op_desc, "_AllShape", all_shape); |
|
|
|
bool dynamic_flag = false; |
|
|
|
if (CheckShapeReset(op_desc, dynamic_flag) == SUCCESS && dynamic_flag) { |
|
|
|
CheckShapeReset(op_desc, dynamic_flag); |
|
|
|
if (all_shape && dynamic_flag) { |
|
|
|
GELOGD("Get aicpu all_shape kernel!"); |
|
|
|
vector<GeTensor> inputs_dynamic; |
|
|
|
vector<GeTensor> outputs_dynamic; |
|
|
|
GE_CHK_STATUS_RET_NOLOG(ResetTensorVecShape(inputs, inputs_dynamic)); |
|
|
|