From a87922d6544d146f8e2858698a2b8d2b7386d29c Mon Sep 17 00:00:00 2001 From: "gengchao4@huawei.com" Date: Thu, 14 Jan 2021 17:30:14 +0800 Subject: [PATCH] modify tf_data_parser check --- parser/tensorflow/tensorflow_data_parser.cc | 5 +++-- 1 file changed, 3 insertions(+), 2 deletions(-) diff --git a/parser/tensorflow/tensorflow_data_parser.cc b/parser/tensorflow/tensorflow_data_parser.cc index 965d73b..e2839e9 100644 --- a/parser/tensorflow/tensorflow_data_parser.cc +++ b/parser/tensorflow/tensorflow_data_parser.cc @@ -81,8 +81,9 @@ Status TensorFlowDataParser::ParseInputFromModel(const Message *op_src, ge::OpDe } if (!TensorFlowUtil::FindAttrValue(node, TENSORFLOW_ATTR_SHAPE, attr_value)) { - GELOGE(domi::PARAM_INVALID, "input data node %s do not find shape.", node->name().c_str()); - return domi::PARAM_INVALID; + // in some case, data could be without shape and is updated by `input_shape` option in following process + GELOGW("input data node %s do not find shape.", node->name().c_str()); + return SUCCESS; } // Check shape attribute must be shape