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