For aicpu all_shape compile.

pull/1016/head
unknown 4 years ago
parent 975dddc0d5
commit f640f729c9

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

Loading…
Cancel
Save