From 60db8cd31d8acfef4d1916d57f971b8927d2f442 Mon Sep 17 00:00:00 2001 From: l00444296 Date: Thu, 10 Dec 2020 15:41:02 +0800 Subject: [PATCH] Feature: reset shape of dynamic single op --- ge/generator/ge_generator.cc | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/ge/generator/ge_generator.cc b/ge/generator/ge_generator.cc index 7140170e..79d39320 100644 --- a/ge/generator/ge_generator.cc +++ b/ge/generator/ge_generator.cc @@ -595,7 +595,7 @@ Status GeGenerator::CheckForSingleOp(OpDescPtr &op_desc, const vector Status GeGenerator::BuildSingleOp(OpDescPtr &op_desc, const vector &inputs, const vector &outputs, const string &model_file_name, OpEngineType engine_type, ModelBufferData &model_buff, bool is_offline) { - if (is_offline) { + if (!is_offline) { (void)AttrUtils::SetBool(op_desc, ATTR_DYNAMIC_SHAPE_SINGLE_AICPU, true); } if (CheckForSingleOp(op_desc, inputs, outputs) != SUCCESS) {