!453 atc json remove unknow shape check

From: @wangxiaotian22
Reviewed-by: @xchu42,@ji_chen
Signed-off-by: @ji_chen
pull/453/MERGE
mindspore-ci-bot 4 years ago committed by Gitee
commit dfc43fca00

@ -95,8 +95,8 @@ static void ParseAtcParms(const std::map<std::string, std::string> &atc_params,
}
}
static Status CheckInputShapeNode(const ComputeGraphPtr &graph, const bool is_dynamic_input) {
if (!is_dynamic_input) {
static Status CheckInputShapeNode(const ComputeGraphPtr &graph, const bool is_dynamic_input, RunMode run_mode) {
if (!is_dynamic_input && run_mode != MODEL_TO_JSON) {
for (auto node : graph->GetDirectNode()) {
if (node->GetType() == DATA) {
auto data_op_desc = node->GetOpDesc();
@ -793,7 +793,7 @@ FMK_FUNC_HOST_VISIBILITY Status ParseGraph(ge::Graph &graph, const std::map<stri
compute_graph = GraphUtils::GetComputeGraph(graph);
GE_RETURN_IF_ERROR(CheckInputFp16Nodes(compute_graph, input_fp16_nodes, is_input_adjust_hw_layout));
GE_RETURN_IF_ERROR(CheckInputShapeNode(compute_graph, is_dynamic_input));
GE_RETURN_IF_ERROR(CheckInputShapeNode(compute_graph, is_dynamic_input, run_mode));
std::string compress_weight_conf;
ParseAtcParms(atc_params, "compress_weight_conf", compress_weight_conf);

Loading…
Cancel
Save