diff --git a/ge/generator/ge_generator.cc b/ge/generator/ge_generator.cc index 52dfa65d..6c0fb6e5 100644 --- a/ge/generator/ge_generator.cc +++ b/ge/generator/ge_generator.cc @@ -726,7 +726,7 @@ Status GeGenerator::BuildSingleOp(OpDescPtr &op_desc, const vector &in bool dynamic_flag = false; (void)AttrUtils::GetBool(op_desc, kAicpuAllshape, all_shape); CheckShapeReset(op_desc, dynamic_flag); - if (dynamic_flag && all_shape) { + if (dynamic_flag || all_shape) { GELOGD("Get aicpu all_shape kernel!"); vector inputs_dynamic; vector outputs_dynamic;